Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,6 @@ print(f"Running on: {DEVICE.upper()} | dtype: {DTYPE}")
|
|
| 12 |
# ββ Model ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 13 |
MODEL_REPO = "cyberdelia/CyberRealisticPony"
|
| 14 |
MODEL_FILE = "CyberRealisticPony_V16.0_FP16.safetensors"
|
| 15 |
-
MODEL_URL = f"https://huggingface.co/{MODEL_REPO}/resolve/main/{MODEL_FILE}"
|
| 16 |
|
| 17 |
# ββ Pony quality tags (added automatically β do not type these in your prompt) β
|
| 18 |
PONY_POS_PREFIX = "score_9, score_8_up, score_7_up, "
|
|
@@ -22,9 +21,9 @@ PONY_NEG_PREFIX = "score_6, score_5, score_4, "
|
|
| 22 |
print("Loading model β this may take several minutes on first start...")
|
| 23 |
|
| 24 |
pipe = StableDiffusionXLPipeline.from_single_file(
|
| 25 |
-
|
|
|
|
| 26 |
torch_dtype=DTYPE,
|
| 27 |
-
use_safetensors=True,
|
| 28 |
)
|
| 29 |
|
| 30 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(
|
|
|
|
| 12 |
# ββ Model ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 13 |
MODEL_REPO = "cyberdelia/CyberRealisticPony"
|
| 14 |
MODEL_FILE = "CyberRealisticPony_V16.0_FP16.safetensors"
|
|
|
|
| 15 |
|
| 16 |
# ββ Pony quality tags (added automatically β do not type these in your prompt) β
|
| 17 |
PONY_POS_PREFIX = "score_9, score_8_up, score_7_up, "
|
|
|
|
| 21 |
print("Loading model β this may take several minutes on first start...")
|
| 22 |
|
| 23 |
pipe = StableDiffusionXLPipeline.from_single_file(
|
| 24 |
+
MODEL_REPO,
|
| 25 |
+
weight_name=MODEL_FILE,
|
| 26 |
torch_dtype=DTYPE,
|
|
|
|
| 27 |
)
|
| 28 |
|
| 29 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(
|