Spaces:
Runtime error
Runtime error
fix lora loading
Browse files
model.py
CHANGED
|
@@ -203,7 +203,7 @@ class Model:
|
|
| 203 |
).to(self.device)
|
| 204 |
self.pipe.enable_xformers_memory_efficient_attention()
|
| 205 |
self.pipe.load_lora_weights("stabilityai/stable-diffusion-xl-base-1.0", weight_name="sd_xl_offset_example-lora_1.0.safetensors")
|
| 206 |
-
pipe.fuse_lora(lora_scale=0.4)
|
| 207 |
else:
|
| 208 |
self.preprocessor = None # type: ignore
|
| 209 |
self.pipe = None
|
|
|
|
| 203 |
).to(self.device)
|
| 204 |
self.pipe.enable_xformers_memory_efficient_attention()
|
| 205 |
self.pipe.load_lora_weights("stabilityai/stable-diffusion-xl-base-1.0", weight_name="sd_xl_offset_example-lora_1.0.safetensors")
|
| 206 |
+
self.pipe.fuse_lora(lora_scale=0.4)
|
| 207 |
else:
|
| 208 |
self.preprocessor = None # type: ignore
|
| 209 |
self.pipe = None
|