Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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["
|
183 |
else:
|
184 |
-
config_l["
|
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 |
|