Spaces:
Runtime error
Runtime error
Update diffusion_webui/diffusion_models/inpaint_app.py
Browse files
diffusion_webui/diffusion_models/inpaint_app.py
CHANGED
|
@@ -12,7 +12,7 @@ class StableDiffusionInpaintGenerator:
|
|
| 12 |
def load_model(self, stable_model_path):
|
| 13 |
if self.pipe is None or self.pipe.model_name != stable_model_path:
|
| 14 |
self.pipe = DiffusionPipeline.from_pretrained(
|
| 15 |
-
|
| 16 |
)
|
| 17 |
self.pipe.to("cuda")
|
| 18 |
self.pipe.enable_xformers_memory_efficient_attention()
|
|
|
|
| 12 |
def load_model(self, stable_model_path):
|
| 13 |
if self.pipe is None or self.pipe.model_name != stable_model_path:
|
| 14 |
self.pipe = DiffusionPipeline.from_pretrained(
|
| 15 |
+
stable_model_path, revision="fp16", torch_dtype=torch.float16
|
| 16 |
)
|
| 17 |
self.pipe.to("cuda")
|
| 18 |
self.pipe.enable_xformers_memory_efficient_attention()
|