Spaces:

tokeron
/

DiffusionLens

Sleeping

tokeron commited on Apr 17

Commit

b1e8435

•

1 Parent(s): 5fbbdb5

Update diffusion_lens.py

Files changed (1) hide show

diffusion_lens.py CHANGED Viewed

@@ -4,5 +4,6 @@ pipeline = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-
 pipeline.scheduler = EulerDiscreteScheduler.from_config(pipeline.scheduler.config)
 def get_images(prompt, skip_layers=0):
-    images = pipeline(prompt, skip_layers=skip_layers, num_images_per_prompt=1)
     return images

 pipeline.scheduler = EulerDiscreteScheduler.from_config(pipeline.scheduler.config)
 def get_images(prompt, skip_layers=0):
+    pipeline_output = pipeline(prompt, skip_layers=skip_layers, num_images_per_prompt=1)
+    images = pipeline_output.images
     return images