Shuttle Jaguar

🚀 穿梭美洲豹模型

穿梭美洲豹（Shuttle Jaguar）是一款文本到圖像的AI模型，它能依據文本提示，僅通過四個步驟就生成極具美感、富有電影質感且逼真的圖像。該模型遵循Apache 2許可協議。

你可以加入我們的 Discord / Telegram 社群，獲取最新的更新、消息等內容。

這些模型變體提供了不同的精度級別和格式，針對不同的硬件能力和使用場景進行了優化：

• bfloat16
• fp8
• GGUF

🚀 快速開始

📦 安裝指南

使用 🧨 Diffusers 庫

安裝或升級 diffusers 庫：

pip install -U diffusers

💻 使用示例

使用 🧨 Diffusers 庫

import torch
from diffusers import DiffusionPipeline

# 從預訓練模型加載擴散管道，使用 bfloat16 作為張量類型。
pipe = DiffusionPipeline.from_pretrained(
    "shuttleai/shuttle-jaguar", torch_dtype=torch.bfloat16
).to("cuda")

# 若需要，取消註釋以下行以通過將模型卸載到 CPU 來節省顯存。
# pipe.enable_model_cpu_offload()

# 取消註釋以下行以在兼容的 GPU 上啟用 torch.compile 以提升性能。
# 注意，這可能會顯著增加加載時間。
# pipe.transformer.to(memory_format=torch.channels_last)
# pipe.transformer = torch.compile(
#     pipe.transformer, mode="max-autotune", fullgraph=True
# )

# 設置圖像生成的提示詞。
prompt = "A cat holding a sign that says hello world"

# 使用擴散管道生成圖像。
image = pipe(
    prompt,
    height=1024,
    width=1024,
    guidance_scale=3.5,
    num_inference_steps=4,
    max_sequence_length=256,
    # 若需要可復現的結果，取消註釋以下行以使用手動種子。
    # generator=torch.Generator("cpu").manual_seed(0)
).images[0]

# 保存生成的圖像。
image.save("shuttle.png")

更多信息請查看 diffusers 文檔。

通過 API 使用模型

你可以通過ShuttleAI的API使用穿梭美洲豹模型：

• ShuttleAI
• ShuttleAI 文檔

使用 ComfyUI 運行模型

若要使用 ComfyUI 對穿梭美洲豹模型進行本地推理，你可以使用這個 safetensors 文件。

📄 許可證

本模型遵循 Apache 2 許可協議。