AbstractPhil commited on
Commit
65d629b
·
verified ·
1 Parent(s): 9da4d4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -179,9 +179,9 @@ def infer(prompt, negative_prompt, adapter_l_file, adapter_g_file, strength, del
179
 
180
  # Load and apply adapters
181
  if(adapter_l_file == "t5-vit-l-14-dual_shunt_booru_13_000_000.safetensors"):
182
- config_l["num_heads"] = 4
183
  else:
184
- config_l["num_heads"] = 12
185
  adapter_l = load_adapter(repo_l, adapter_l_file, config_l, device) if adapter_l_file else None
186
  adapter_g = load_adapter(repo_g, adapter_g_file, config_g, device) if adapter_g_file else None
187
 
 
179
 
180
  # Load and apply adapters
181
  if(adapter_l_file == "t5-vit-l-14-dual_shunt_booru_13_000_000.safetensors"):
182
+ config_l["heads"] = 4
183
  else:
184
+ config_l["heads"] = 12
185
  adapter_l = load_adapter(repo_l, adapter_l_file, config_l, device) if adapter_l_file else None
186
  adapter_g = load_adapter(repo_g, adapter_g_file, config_g, device) if adapter_g_file else None
187