thankfulcarp commited on
Commit
9e04e04
·
1 Parent(s): 32b8238

Load changes

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -51,9 +51,9 @@ def load_and_fuse_pipeline(model_id, lora_filename, pipeline_class, lora_repo_id
51
  return None
52
 
53
  try:
54
- lora_path = hf_hub_download(repo_id=lora_repo_id, filename=lora_filename)
55
- print(f"✅ LoRA downloaded for {model_id} to: {lora_path}")
56
- pipe.load_lora_weights(lora_path, adapter_name="fusionx_lora")
57
  pipe.set_adapters(["fusionx_lora"], adapter_weights=[0.75])
58
  pipe.fuse_lora()
59
  print(f"✅ FusionX LoRA loaded and fused for {model_id} with a weight of 0.75.")
 
51
  return None
52
 
53
  try:
54
+ print(f"✅ Loading LoRA for {model_id}...")
55
+ # Use the more direct loading method from the diffusers library
56
+ pipe.load_lora_weights(lora_repo_id, weight_name=lora_filename, adapter_name="fusionx_lora")
57
  pipe.set_adapters(["fusionx_lora"], adapter_weights=[0.75])
58
  pipe.fuse_lora()
59
  print(f"✅ FusionX LoRA loaded and fused for {model_id} with a weight of 0.75.")