Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -182,9 +182,9 @@ def infer(
|
|
| 182 |
if lora_adapter == "QWEN_MCNL":
|
| 183 |
pipe.set_adapters(["anime"], adapter_weights=[1.0])
|
| 184 |
elif lora_adapter == "Multiple-Angles":
|
| 185 |
-
pipe.set_adapters(["multiple-angles"], adapter_weights=[0
|
| 186 |
elif lora_adapter == "MXXX":
|
| 187 |
-
pipe.set_adapters(["light-restoration"], adapter_weights=[0
|
| 188 |
elif lora_adapter == "Qwen-lora-nsfw":
|
| 189 |
pipe.set_adapters(["relight", "multi-angle-lighting"], adapter_weights=[1.0, 1.0])
|
| 190 |
elif lora_adapter == "Jib":
|
|
@@ -220,11 +220,11 @@ def infer(
|
|
| 220 |
|
| 221 |
return result, seed
|
| 222 |
|
| 223 |
-
@spaces.GPU(duration=
|
| 224 |
def infer_example(input_image, prompt, lora_adapter):
|
| 225 |
input_pil = input_image.convert("RGB")
|
| 226 |
-
guidance_scale =
|
| 227 |
-
steps =
|
| 228 |
result, seed = infer(input_pil, prompt, lora_adapter, 0, True, guidance_scale, steps)
|
| 229 |
return result, seed
|
| 230 |
|
|
|
|
| 182 |
if lora_adapter == "QWEN_MCNL":
|
| 183 |
pipe.set_adapters(["anime"], adapter_weights=[1.0])
|
| 184 |
elif lora_adapter == "Multiple-Angles":
|
| 185 |
+
pipe.set_adapters(["multiple-angles"], adapter_weights=[1.0])
|
| 186 |
elif lora_adapter == "MXXX":
|
| 187 |
+
pipe.set_adapters(["light-restoration"], adapter_weights=[1.0])
|
| 188 |
elif lora_adapter == "Qwen-lora-nsfw":
|
| 189 |
pipe.set_adapters(["relight", "multi-angle-lighting"], adapter_weights=[1.0, 1.0])
|
| 190 |
elif lora_adapter == "Jib":
|
|
|
|
| 220 |
|
| 221 |
return result, seed
|
| 222 |
|
| 223 |
+
@spaces.GPU(duration=60)
|
| 224 |
def infer_example(input_image, prompt, lora_adapter):
|
| 225 |
input_pil = input_image.convert("RGB")
|
| 226 |
+
guidance_scale = 4.0
|
| 227 |
+
steps = 10
|
| 228 |
result, seed = infer(input_pil, prompt, lora_adapter, 0, True, guidance_scale, steps)
|
| 229 |
return result, seed
|
| 230 |
|