Diffusers How to use wsl448/yarn_art_falqon_sana_block_int8_svd_lr2e-4 with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("wsl448/yarn_art_falqon_sana_block_int8_svd_lr2e-4", dtype=torch.bfloat16, device_map="cuda")
prompt = "a puppy in a pond, yarn art style"
image = pipe(prompt).images[0] Sana How to use wsl448/yarn_art_falqon_sana_block_int8_svd_lr2e-4 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://wsl448/yarn_art_falqon_sana_block_int8_svd_lr2e-4")
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,
) 
