valhalla commited on
Commit
5e63b46
1 Parent(s): be7a25b

fix lora loading

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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