Spaces:

dbaranchuk
/

Scale-wise-Distillation

Running on Zero

dbaranchuk commited on 20 days ago

Commit

951aec2

verified ·

1 Parent(s): 778b7ae

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -47,7 +47,7 @@ def infer(prompt, seed, randomize_seed):
             height=int(scales[0] * 8),
             width=int(scales[0] * 8),
             generator=generator,
-        ).images
     return images

             height=int(scales[0] * 8),
             width=int(scales[0] * 8),
             generator=generator,
+        ).images[0]
     return images