Hugging Face's logo

Spaces:
Ava2lon
/
FLUX
Running on Zero

App Files Community
multimodalart HF Staff commited on
Commit
f7170dc
·
verified ·
1 Parent(s): e1e80db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,8 +48,8 @@ Rules:
48
  Output only the final instruction in plain text and nothing else."""
49
 
50
  # Model repository IDs for 4B
51
- REPO_ID_REGULAR = "diffusers-internal-dev/dummy-1015-4b"
52
- REPO_ID_DISTILLED = "diffusers-internal-dev/dummy-1015-4b-distilled"
53
 
54
  # Load both 4B models
55
  print("Loading 4B Regular model...")
 
48
  Output only the final instruction in plain text and nothing else."""
49
 
50
  # Model repository IDs for 4B
51
+ REPO_ID_REGULAR = "black-forest-labs/FLUX.2-klein-base-4B"
52
+ REPO_ID_DISTILLED = "black-forest-labs/FLUX.2-klein-4B"
53
 
54
  # Load both 4B models
55
  print("Loading 4B Regular model...")