Spaces:
Running on Zero

Ruurd commited on
Commit
3125ce6
·
verified ·
1 Parent(s): cea9bb0

Change standard settings and revert to 8B

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -183,7 +183,7 @@ if is_running_on_spaces():
183
  # Load from Hugging Face Hub
184
  ckpt_path = hf_hub_download(
185
  repo_id="ruurd/tini_model",
186
- filename="diffusion-model-3B.pth",
187
  token=os.getenv("HF_TOKEN")
188
  )
189
  else:
@@ -206,12 +206,12 @@ demo = gr.Interface(
206
  gr.Slider(0.01, 5, value=0.01, step=0.01, label="Pause between iteration ↑ = longer pause"),
207
  gr.Slider(1.0, 20.0, value=1.0, step=0.5, label="Noise decay sharpness: ↓ = more noise in later iterations"),
208
  gr.Slider(0.0, 1.0, value=0.0, step=0.05, label="Clustering: ↑ = more clustered noising"),
209
- gr.Slider(0.0, 1.0, value=0.2, step=0.05, label="Noise start fraction: ↑ = more noise"),
210
  gr.Checkbox(value=False, label="Use confidence-guided noising"),
211
  gr.Checkbox(value=False, label="Use permanent unmasking"),
212
  gr.Slider(0.01, 1.0, value=0.01, step=0.01, label="Noise clipping: ↓ = more confidence guidance"),
213
- gr.Slider(1, 1000, value = 100, step = 1, label = "Top-p: ↑ = more random answers"),
214
- gr.Slider(0.0, 1.0, value = 0.9, step = 0.01, label = "Top-k: ↑ = more random answers")
215
  ],
216
  outputs=[gr.HTML(label="Diffusion Output")],
217
  title="Diffusion Language Model Chat",
 
183
  # Load from Hugging Face Hub
184
  ckpt_path = hf_hub_download(
185
  repo_id="ruurd/tini_model",
186
+ filename="diffusion-model-8B.pth",
187
  token=os.getenv("HF_TOKEN")
188
  )
189
  else:
 
206
  gr.Slider(0.01, 5, value=0.01, step=0.01, label="Pause between iteration ↑ = longer pause"),
207
  gr.Slider(1.0, 20.0, value=1.0, step=0.5, label="Noise decay sharpness: ↓ = more noise in later iterations"),
208
  gr.Slider(0.0, 1.0, value=0.0, step=0.05, label="Clustering: ↑ = more clustered noising"),
209
+ gr.Slider(0.0, 1.0, value=0.5, step=0.05, label="Noise start fraction: ↑ = more noise"),
210
  gr.Checkbox(value=False, label="Use confidence-guided noising"),
211
  gr.Checkbox(value=False, label="Use permanent unmasking"),
212
  gr.Slider(0.01, 1.0, value=0.01, step=0.01, label="Noise clipping: ↓ = more confidence guidance"),
213
+ gr.Slider(1, 1000, value = 3, step = 1, label = "Top-p: ↑ = more random answers"),
214
+ gr.Slider(0.0, 1.0, value = 1.0, step = 0.01, label = "Top-k: ↑ = more random answers")
215
  ],
216
  outputs=[gr.HTML(label="Diffusion Output")],
217
  title="Diffusion Language Model Chat",