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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -17,9 +17,13 @@ LORA_WRONG_REPO = "minimaxir/sdxl-wrong-lora" # [web:183]
17
  LORA_WRONG_ADAPTER = "wrong_prompt"
18
 
19
  # LoRA visages / détails
20
- LORA_FACE_REPO = "akash-guptag/Detailers_By_Stable_Yogi" # [web:60][web:63]
21
  LORA_FACE_ADAPTER = "face_detail"
22
 
 
 
 
 
23
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
24
  DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32
25
 
@@ -50,6 +54,8 @@ pipe.load_lora_weights(LORA_WRONG_REPO, adapter_name=LORA_WRONG_ADAPTER)
50
  print("Chargement LoRA face/detail (Stable Yogi)...")
51
  pipe.load_lora_weights(LORA_FACE_REPO, adapter_name=LORA_FACE_ADAPTER)
52
 
 
 
53
  os.makedirs("outputs", exist_ok=True)
54
 
55
 
@@ -99,8 +105,8 @@ def generate(
99
  final_negative = "wrong, blurry, low quality, deformed, bad anatomy, extra limbs"
100
 
101
  # On combine les deux LoRA : WRONG à 1.0, face en slider
102
- adapters = [LORA_WRONG_ADAPTER, LORA_FACE_ADAPTER]
103
- weights = [1.0, float(face_weight)]
104
  pipe.set_adapters(adapters, adapter_weights=weights)
105
 
106
  try:
 
17
  LORA_WRONG_ADAPTER = "wrong_prompt"
18
 
19
  # LoRA visages / détails
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"
28
  DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32
29
 
 
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
 
61
 
 
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
 
112
  try: