Exched commited on
Commit
81da150
·
1 Parent(s): 221414e

use exched/checkpoint_n_lora dataset

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -60,7 +60,7 @@ R2_NAMESPACE = "s01"
60
  # Model registry
61
  # =============================================================================
62
  # Two pipelines are exposed through a single UI via the selector below. Both now
63
- # run custom single-file checkpoints pulled from the nsfwalex/checkpoint_n_lora
64
  # dataset (populated by the model_downloader Space):
65
  # * Moody Pro Mix -> Z-Image-Turbo finetune (fast, guidance-free DiT)
66
  # * One Obsession -> Illustrious / SDXL anime model with Compel weighting
@@ -84,7 +84,7 @@ def _supports_negative(model_name):
84
 
85
  # Custom checkpoints live in this HF dataset; HF_TOKEN (a Space secret) grants
86
  # read access. Paths mirror the model_downloader index.json layout.
87
- CHECKPOINT_DATASET = "nsfwalex/checkpoint_n_lora"
88
  # Per-model checkpoint paths in the dataset live in the IMAGE_MODELS registry.
89
  # Base Z-Image repo supplies the VAE / text encoder / tokenizer / scheduler and
90
  # the transformer config; the single-file checkpoint only carries the DiT.
@@ -202,7 +202,7 @@ def _apply_prefix(prompt, prefix):
202
  # Image-model registry + startup loader.
203
  #
204
  # Each entry says which pipeline *family* loads the model, its checkpoint in the
205
- # `nsfwalex/checkpoint_n_lora` dataset, sane sampler defaults, and whether it
206
  # honors a negative prompt. Adding a model = one entry here + uploading its
207
  # checkpoint (via the model_downloader Space); no other code change.
208
  #
 
60
  # Model registry
61
  # =============================================================================
62
  # Two pipelines are exposed through a single UI via the selector below. Both now
63
+ # run custom single-file checkpoints pulled from the exched/checkpoint_n_lora
64
  # dataset (populated by the model_downloader Space):
65
  # * Moody Pro Mix -> Z-Image-Turbo finetune (fast, guidance-free DiT)
66
  # * One Obsession -> Illustrious / SDXL anime model with Compel weighting
 
84
 
85
  # Custom checkpoints live in this HF dataset; HF_TOKEN (a Space secret) grants
86
  # read access. Paths mirror the model_downloader index.json layout.
87
+ CHECKPOINT_DATASET = "exched/checkpoint_n_lora"
88
  # Per-model checkpoint paths in the dataset live in the IMAGE_MODELS registry.
89
  # Base Z-Image repo supplies the VAE / text encoder / tokenizer / scheduler and
90
  # the transformer config; the single-file checkpoint only carries the DiT.
 
202
  # Image-model registry + startup loader.
203
  #
204
  # Each entry says which pipeline *family* loads the model, its checkpoint in the
205
+ # `exched/checkpoint_n_lora` dataset, sane sampler defaults, and whether it
206
  # honors a negative prompt. Adding a model = one entry here + uploading its
207
  # checkpoint (via the model_downloader Space); no other code change.
208
  #