Spaces:
Runtime error
Runtime error
load lora
Browse files
model.py
CHANGED
@@ -202,6 +202,8 @@ class Model:
|
|
202 |
variant="fp16",
|
203 |
).to(self.device)
|
204 |
self.pipe.enable_xformers_memory_efficient_attention()
|
|
|
|
|
205 |
else:
|
206 |
self.preprocessor = None # type: ignore
|
207 |
self.pipe = None
|
@@ -278,7 +280,7 @@ class Model:
|
|
278 |
image=image,
|
279 |
num_inference_steps=num_inference_steps,
|
280 |
adapter_conditioning_scale=adapter_conditioning_scale,
|
281 |
-
|
282 |
generator=generator,
|
283 |
guidance_scale=guidance_scale,
|
284 |
).images[0]
|
|
|
202 |
variant="fp16",
|
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
|
|
|
280 |
image=image,
|
281 |
num_inference_steps=num_inference_steps,
|
282 |
adapter_conditioning_scale=adapter_conditioning_scale,
|
283 |
+
adapter_conditioning_factor=cond_tau,
|
284 |
generator=generator,
|
285 |
guidance_scale=guidance_scale,
|
286 |
).images[0]
|