hunyuanvideo-community
/

HunyuanImage-2.1-Distilled-Diffusers

HunyuanImagePipeline

Model card Files Files and versions

YiYiXu commited on Oct 23

Commit

630179c

·

verified ·

1 Parent(s): 31eaaf3

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -5,7 +5,7 @@ import torch
 device = "cuda:0"
 dtype = torch.bfloat16
-repo = "YiYiXu/HunyuanImage-2.1-Distilled-Diffusers"
 pipe = HunyuanImagePipeline.from_pretrained(repo, torch_dtype=dtype)
 pipe = pipe.to(device)
@@ -15,7 +15,7 @@ generator = torch.Generator(device=device).manual_seed(649151)
 out = pipe(
     prompt,
     num_inference_steps=8,
-    guidance_scale =3.5,
     height=2048,
     width=2048,
     generator=generator,

 device = "cuda:0"
 dtype = torch.bfloat16
+repo = "hunyuanvideo-community/HunyuanImage-2.1-Distilled-Diffusers"
 pipe = HunyuanImagePipeline.from_pretrained(repo, torch_dtype=dtype)
 pipe = pipe.to(device)
 out = pipe(
     prompt,
     num_inference_steps=8,
+    distilled_guidance_scale =3.5,
     height=2048,
     width=2048,
     generator=generator,