Spaces:
Running
on
Zero
Running
on
Zero
Update model.py
Browse files
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'
|