Spaces:
Runtime error
Runtime error
Update diffusion_webui/diffusion_models/text2img_app.py
Browse files
diffusion_webui/diffusion_models/text2img_app.py
CHANGED
@@ -20,7 +20,7 @@ class StableDiffusionText2ImageGenerator:
|
|
20 |
):
|
21 |
if self.pipe is None or self.pipe.model_name != stable_model_path or self.pipe.scheduler_name != scheduler:
|
22 |
self.pipe = StableDiffusionPipeline.from_pretrained(
|
23 |
-
|
24 |
)
|
25 |
|
26 |
self.pipe = get_scheduler(pipe=self.pipe, scheduler=scheduler)
|
|
|
20 |
):
|
21 |
if self.pipe is None or self.pipe.model_name != stable_model_path or self.pipe.scheduler_name != scheduler:
|
22 |
self.pipe = StableDiffusionPipeline.from_pretrained(
|
23 |
+
stable_model_path, safety_checker=None, torch_dtype=torch.float16
|
24 |
)
|
25 |
|
26 |
self.pipe = get_scheduler(pipe=self.pipe, scheduler=scheduler)
|