ACE-Plus

Running on Zero

chaojiemao commited on Jan 14

Commit

3bd6527

verified ·

1 Parent(s): 220ccd7

Update inference/ace_plus_diffusers.py

Files changed (1) hide show

inference/ace_plus_diffusers.py CHANGED Viewed

@@ -108,6 +108,8 @@ class ACEPlusDiffuserInference():
             max_sequence_length=512,
             generator=generator
         ).images[0]
         return self.image_processor.postprocess(image, slice_w, out_w, out_h), seed

             max_sequence_length=512,
             generator=generator
         ).images[0]
+        if lora_path is not None:
+            self.pipe.unload_lora_weights()
         return self.image_processor.postprocess(image, slice_w, out_w, out_h), seed