Update app.py
Browse files
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"
|
| 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:
|