AbstractPhil commited on
Commit
9da4d4e
·
verified ·
1 Parent(s): 07930c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -178,6 +178,10 @@ 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
  adapter_l = load_adapter(repo_l, adapter_l_file, config_l, device) if adapter_l_file else None
182
  adapter_g = load_adapter(repo_g, adapter_g_file, config_g, device) if adapter_g_file else None
183
 
 
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["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