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
|