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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -178,7 +178,7 @@ def infer(prompt, negative_prompt, adapter_l_file, adapter_g_file, strength, del
178
  clip_embeds = encode_sdxl_prompt(pipe, prompt, negative_prompt, device)
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
 
178
  clip_embeds = encode_sdxl_prompt(pipe, prompt, negative_prompt, device)
179
 
180
  # Load and apply adapters
181
+ if(adapter_l_file == "t5-vit-l-14-dual_shunt_booru_13_000_000.safetensors" or adapter_l_file == "t5-vit-l-14-dual_shunt_booru_51_200_000.safetensors"):
182
  config_l["heads"] = 4
183
  else:
184
  config_l["heads"] = 12