GiorgioV commited on
Commit
9e7e284
·
verified ·
1 Parent(s): 8351fb1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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.8])
186
  elif lora_adapter == "MXXX":
187
- pipe.set_adapters(["light-restoration"], adapter_weights=[0.8])
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=120)
224
  def infer_example(input_image, prompt, lora_adapter):
225
  input_pil = input_image.convert("RGB")
226
- guidance_scale = 1.0
227
- steps = 20
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