tomiconic commited on
Commit
f037026
Β·
verified Β·
1 Parent(s): 4f12d22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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
- MODEL_URL,
 
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(