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