ironjr commited on
Commit
7a2011f
β€’
1 Parent(s): 93344b8

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -173,7 +173,7 @@ class StableMultiDiffusionSDXLPipeline(nn.Module):
173
  self.pipe = StableDiffusionXLPipeline.from_pretrained(model_key, variant=variant, torch_dtype=self.dtype).to(self.device)
174
  self.pipe.load_lora_weights(hf_hub_download(lightning_repo, lora_ckpt), adapter_name='lightning')
175
  self.pipe.set_adapters(["lightning"], adapter_weights=[lora_weight])
176
- self.pipe.fuse_lora()
177
  else:
178
  model_key = 'stabilityai/stable-diffusion-xl-base-1.0'
179
  variant = 'fp16'
 
173
  self.pipe = StableDiffusionXLPipeline.from_pretrained(model_key, variant=variant, torch_dtype=self.dtype).to(self.device)
174
  self.pipe.load_lora_weights(hf_hub_download(lightning_repo, lora_ckpt), adapter_name='lightning')
175
  self.pipe.set_adapters(["lightning"], adapter_weights=[lora_weight])
176
+ # self.pipe.fuse_lora()
177
  else:
178
  model_key = 'stabilityai/stable-diffusion-xl-base-1.0'
179
  variant = 'fp16'