Update app.py
Browse files
app.py
CHANGED
|
@@ -24,6 +24,7 @@ import numpy as np
|
|
| 24 |
|
| 25 |
# 1. Base Model - Using reliable SDXL models (avoiding incomplete "bait" models)
|
| 26 |
BASE_MODELS = {
|
|
|
|
| 27 |
"sdxl_base": "stabilityai/stable-diffusion-xl-base-1.0", # Most reliable choice
|
| 28 |
"realistic_vision": "SG161222/RealVisXL_V4.0", # High-quality realistic model
|
| 29 |
"anime_xl": "Linaqruf/animagine-xl-3.1", # Popular anime-style SDXL
|
|
@@ -32,7 +33,7 @@ BASE_MODELS = {
|
|
| 32 |
}
|
| 33 |
|
| 34 |
# Current model selection (change this to switch models)
|
| 35 |
-
CURRENT_MODEL_KEY = "
|
| 36 |
BASE_MODEL = BASE_MODELS[CURRENT_MODEL_KEY]
|
| 37 |
|
| 38 |
# 2. Fixed LoRAs (Auto-loaded, not user-selectable) - Using actual LoRA models
|
|
|
|
| 24 |
|
| 25 |
# 1. Base Model - Using reliable SDXL models (avoiding incomplete "bait" models)
|
| 26 |
BASE_MODELS = {
|
| 27 |
+
"realistic":"John6666/wai-nsfw-illustrious-v80-sdxl",
|
| 28 |
"sdxl_base": "stabilityai/stable-diffusion-xl-base-1.0", # Most reliable choice
|
| 29 |
"realistic_vision": "SG161222/RealVisXL_V4.0", # High-quality realistic model
|
| 30 |
"anime_xl": "Linaqruf/animagine-xl-3.1", # Popular anime-style SDXL
|
|
|
|
| 33 |
}
|
| 34 |
|
| 35 |
# Current model selection (change this to switch models)
|
| 36 |
+
CURRENT_MODEL_KEY = "realistic" # Changed to working model
|
| 37 |
BASE_MODEL = BASE_MODELS[CURRENT_MODEL_KEY]
|
| 38 |
|
| 39 |
# 2. Fixed LoRAs (Auto-loaded, not user-selectable) - Using actual LoRA models
|