Instructions to use irotem98/sana_bullet with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use irotem98/sana_bullet with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("irotem98/sana_bullet", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Sana
How to use irotem98/sana_bullet with Sana:
# Load the model and infer image from text import torch from app.sana_pipeline import SanaPipeline from torchvision.utils import save_image sana = SanaPipeline("configs/sana_config/1024ms/Sana_1600M_img1024.yaml") sana.from_pretrained("hf://irotem98/sana_bullet") image = sana( prompt='a cyberpunk cat with a neon sign that says "Sana"', height=1024, width=1024, guidance_scale=5.0, pag_guidance_scale=2.0, num_inference_steps=18, ) - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
sana_bullet
Fast finetuned one-step Sana checkpoint for the standalone sana_bullet GPU inference stack.
Links
- GitHub repo: rotem154154/sana_bullet
- Model repo: irotem98/sana_bullet
One-Line Generation
# pip install "git+https://github.com/rotem154154/sana_bullet.git"
from sana_bullet import generate_image
image = generate_image(
"a cinematic portrait of a fox in soft winter light",
model="irotem98/sana_bullet",
)
image
from_pretrained(...) Loading
# pip install "git+https://github.com/rotem154154/sana_bullet.git"
from sana_bullet import SanaBulletPipeline
pipe = SanaBulletPipeline.from_pretrained("irotem98/sana_bullet")
image = pipe.generate("a retro-futuristic motorcycle speeding through neon rain")
image
Server Usage
pip install "git+https://github.com/rotem154154/sana_bullet.git"
export SANA_BULLET_CHECKPOINT_REPO_ID="irotem98/sana_bullet"
python -m sana_bullet.ui_server
Files
step0003000.pt: finetuned transformer checkpoint
Base Stack
- Base model:
Efficient-Large-Model/Sana_Sprint_1.6B_1024px_diffusers - VAE:
dc-ai/dc-ae-lite-f32c32-diffusers
Notes
- The checkpoint is designed for GPU inference.
- The higher-level Python API lives in the GitHub repo, while this model repo only hosts the weights and model card.
- Downloads last month
- -