From 68e0d489d161829101546f6a78646e3b79c9ef4f Mon Sep 17 00:00:00 2001 From: lawrence-cj Date: Sun, 22 Dec 2024 15:27:03 +0800 Subject: [PATCH] update README.md; Signed-off-by: lawrence-cj --- README.md | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 5e3553c..eded332 100644 --- a/README.md +++ b/README.md @@ -149,9 +149,9 @@ import torch from diffusers import SanaPipeline pipe = SanaPipeline.from_pretrained( - "Efficient-Large-Model/Sana_1600M_1024px_diffusers", - variant="fp16", - torch_dtype=torch.float16, + "Efficient-Large-Model/Sana_1600M_1024px_BF16_diffusers", + variant="bf16", + torch_dtype=torch.bfloat16, ) pipe.to("cuda") @@ -163,7 +163,7 @@ image = pipe( prompt=prompt, height=1024, width=1024, - guidance_scale=5.0, + guidance_scale=4.5, num_inference_steps=20, generator=torch.Generator(device="cuda").manual_seed(42), )[0] @@ -212,7 +212,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") generator = torch.Generator(device=device).manual_seed(42) sana = SanaPipeline("configs/sana_config/1024ms/Sana_1600M_img1024.yaml") -sana.from_pretrained("hf://Efficient-Large-Model/Sana_1600M_1024px/checkpoints/Sana_1600M_1024px.pth") +sana.from_pretrained("hf://Efficient-Large-Model/Sana_1600M_1024px_BF16/checkpoints/Sana_1600M_1024px_BF16.pth") prompt = 'a cyberpunk cat with a neon sign that says "Sana"' image = sana(