Spaces:
Runtime error
Runtime error
Update inference/ace_plus_diffusers.py
Browse files
inference/ace_plus_diffusers.py
CHANGED
|
@@ -108,6 +108,8 @@ class ACEPlusDiffuserInference():
|
|
| 108 |
max_sequence_length=512,
|
| 109 |
generator=generator
|
| 110 |
).images[0]
|
|
|
|
|
|
|
| 111 |
return self.image_processor.postprocess(image, slice_w, out_w, out_h), seed
|
| 112 |
|
| 113 |
|
|
|
|
| 108 |
max_sequence_length=512,
|
| 109 |
generator=generator
|
| 110 |
).images[0]
|
| 111 |
+
if lora_path is not None:
|
| 112 |
+
self.pipe.unload_lora_weights()
|
| 113 |
return self.image_processor.postprocess(image, slice_w, out_w, out_h), seed
|
| 114 |
|
| 115 |
|