Spaces:

yeq6x
/

Image2Body_gradio

Runtime error

yeq6x commited on Jul 11, 2024

Commit

76f40f6

1 Parent(s): 20c499b

test lora load

Files changed (1) hide show

scripts/process_utils.py CHANGED Viewed

@@ -52,7 +52,7 @@ def initialize(_use_local=False, use_gpu=False, use_dotenv=False):
 def load_lora(pipeline, lora_path, alpha=0.75):
     pipeline.load_lora_weights(lora_path)
-    # pipeline.fuse_lora(lora_scale=alpha)
 def initialize_sotai_model():
     global device, torch_dtype

 def load_lora(pipeline, lora_path, alpha=0.75):
     pipeline.load_lora_weights(lora_path)
+    pipeline.fuse_lora(lora_scale=alpha)
 def initialize_sotai_model():
     global device, torch_dtype