Leteint commited on
Commit
24ce341
·
verified ·
1 Parent(s): 9ea5425

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -20,8 +20,9 @@ LORA_WRONG_ADAPTER = "wrong_prompt"
20
  LORA_FACE_REPO = "akash-guptag/Detailers_By_Stable_Yogi"
21
  LORA_FACE_ADAPTER = "face_detail"
22
 
23
- # Load the LoRA
24
- pipe.load_lora_weights('ntc-ai/SDXL-LoRA-slider.sexy', weight_name='sexy.safetensors', adapter_name="sexy")
 
25
 
26
 
27
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
@@ -54,7 +55,7 @@ pipe.load_lora_weights(LORA_WRONG_REPO, adapter_name=LORA_WRONG_ADAPTER)
54
  print("Chargement LoRA face/detail (Stable Yogi)...")
55
  pipe.load_lora_weights(LORA_FACE_REPO, adapter_name=LORA_FACE_ADAPTER)
56
 
57
- pipe.load_lora_weights("ntc-ai/SDXL-LoRA-slider.sexy", adapter_name="sexy.safetensors")
58
 
59
  os.makedirs("outputs", exist_ok=True)
60
 
@@ -105,7 +106,7 @@ def generate(
105
  final_negative = "wrong, blurry, low quality, deformed, bad anatomy, extra limbs"
106
 
107
  # On combine les deux LoRA : WRONG à 1.0, face en slider
108
- adapters = [LORA_WRONG_ADAPTER, LORA_FACE_ADAPTER, "ntc-ai/SDXL-LoRA-slider.sexy"]
109
  weights = [1.0, float(face_weight), 3.0]
110
  pipe.set_adapters(adapters, adapter_weights=weights)
111
 
 
20
  LORA_FACE_REPO = "akash-guptag/Detailers_By_Stable_Yogi"
21
  LORA_FACE_ADAPTER = "face_detail"
22
 
23
+ LORA_SEXY_REPO = "ntc-ai/SDXL-LoRA-slider.sexy"
24
+ LORA_SEXY_ADAPTER ="sexy"
25
+
26
 
27
 
28
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
55
  print("Chargement LoRA face/detail (Stable Yogi)...")
56
  pipe.load_lora_weights(LORA_FACE_REPO, adapter_name=LORA_FACE_ADAPTER)
57
 
58
+ pipe.load_lora_weights(LORA_SEXY_REPO, adapter_name=LORA_SEXY_ADAPTER)
59
 
60
  os.makedirs("outputs", exist_ok=True)
61
 
 
106
  final_negative = "wrong, blurry, low quality, deformed, bad anatomy, extra limbs"
107
 
108
  # On combine les deux LoRA : WRONG à 1.0, face en slider
109
+ adapters = [LORA_WRONG_ADAPTER, LORA_FACE_ADAPTER, LORA_SEXY_ADAPTER]
110
  weights = [1.0, float(face_weight), 3.0]
111
  pipe.set_adapters(adapters, adapter_weights=weights)
112