Update app.py
Browse files
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 |
-
|
| 24 |
-
|
|
|
|
| 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(
|
| 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,
|
| 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 |
|