Spaces:
Running
on
Zero
Running
on
Zero
fix: weird typo
Browse files- diffusion.py +4 -6
diffusion.py
CHANGED
@@ -31,11 +31,9 @@ class ImageGenerator:
|
|
31 |
use_safetensors=True,
|
32 |
add_watermarker=False,
|
33 |
)
|
34 |
-
self.pipe.
|
35 |
-
|
36 |
-
|
37 |
-
subfolder="scheduler",
|
38 |
-
)
|
39 |
)
|
40 |
|
41 |
# xformers
|
@@ -68,4 +66,4 @@ class ImageGenerator:
|
|
68 |
width=width,
|
69 |
num_inference_steps=num_inference_steps,
|
70 |
guidance_scale=guidance_scale,
|
71 |
-
).images
|
|
|
31 |
use_safetensors=True,
|
32 |
add_watermarker=False,
|
33 |
)
|
34 |
+
self.pipe.scheduler = EulerAncestralDiscreteScheduler.from_pretrained(
|
35 |
+
model_name,
|
36 |
+
subfolder="scheduler",
|
|
|
|
|
37 |
)
|
38 |
|
39 |
# xformers
|
|
|
66 |
width=width,
|
67 |
num_inference_steps=num_inference_steps,
|
68 |
guidance_scale=guidance_scale,
|
69 |
+
).images
|