GiorgioV commited on
Commit
2bcb8f1
·
verified ·
1 Parent(s): 8c9602c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -192,7 +192,7 @@ def infer(
192
  elif lora_adapter == "Real PS":
193
  pipe.set_adapters(["light-restoration"], adapter_weights=[1.0])
194
  elif lora_adapter == "Qwen-lora-nsfw":
195
- pipe.set_adapters(["relight", "edit-skin", "upscale-image"], adapter_weights=[1., 1., 0.5])
196
  elif lora_adapter == "Yarn":
197
  pipe.set_adapters(["multi-angle-lighting"], adapter_weights=[1.0])
198
  elif lora_adapter == "A2R_2509_Base":
@@ -200,7 +200,7 @@ def infer(
200
  elif lora_adapter == "Meta4":
201
  pipe.set_adapters(["next-scene"], adapter_weights=[1.0])
202
  elif lora_adapter == "V14-10":
203
- pipe.set_adapters(["upscale-image"], adapter_weights=[0.6])
204
 
205
  if randomize_seed:
206
  seed = random.randint(0, MAX_SEED)
 
192
  elif lora_adapter == "Real PS":
193
  pipe.set_adapters(["light-restoration"], adapter_weights=[1.0])
194
  elif lora_adapter == "Qwen-lora-nsfw":
195
+ pipe.set_adapters(["relight", "edit-skin", "upscale-image"], adapter_weights=[1., 0.8, 0.4])
196
  elif lora_adapter == "Yarn":
197
  pipe.set_adapters(["multi-angle-lighting"], adapter_weights=[1.0])
198
  elif lora_adapter == "A2R_2509_Base":
 
200
  elif lora_adapter == "Meta4":
201
  pipe.set_adapters(["next-scene"], adapter_weights=[1.0])
202
  elif lora_adapter == "V14-10":
203
+ pipe.set_adapters(["upscale-image"], adapter_weights=[0.5])
204
 
205
  if randomize_seed:
206
  seed = random.randint(0, MAX_SEED)