hesha
/

wfx

Model card Files Files and versions Community

zwv9 commited on Jan 18

Commit

a299603

•

1 Parent(s): fb27579

fixed

Files changed (1) hide show

wfx.py +1 -1

wfx.py CHANGED Viewed

@@ -102,7 +102,7 @@ class WFX():
         warmup_kwargs = dict(
             prompt='a photo of a cat',
-            height=512,
             width=512,
             num_inference_steps=30,
             generator=torch.Generator(device='cuda:0').manual_seed(0),

         warmup_kwargs = dict(
             prompt='a photo of a cat',
+            height=768,
             width=512,
             num_inference_steps=30,
             generator=torch.Generator(device='cuda:0').manual_seed(0),