Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
|