multimodalart HF Staff commited on
Commit
8e8df28
·
verified ·
1 Parent(s): 2f91598

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -74,7 +74,12 @@ def generate(
74
  # Apply / adjust LoRA strength — load_lora_weights names the adapter "default_0"
75
  adapters = pipe.get_list_adapters()
76
  if adapters:
77
- adapter_names = list(adapters.values())
 
 
 
 
 
78
  else:
79
  adapter_names = ["default_0"]
80
  pipe.set_adapters(adapter_names, adapter_weights=[float(lora_strength)] * len(adapter_names))
 
74
  # Apply / adjust LoRA strength — load_lora_weights names the adapter "default_0"
75
  adapters = pipe.get_list_adapters()
76
  if adapters:
77
+ adapter_names = []
78
+ for v in adapters.values():
79
+ if isinstance(v, list):
80
+ adapter_names.extend(v)
81
+ else:
82
+ adapter_names.append(v)
83
  else:
84
  adapter_names = ["default_0"]
85
  pipe.set_adapters(adapter_names, adapter_weights=[float(lora_strength)] * len(adapter_names))