Spaces:
Running
on
Zero
Running
on
Zero
Change standard settings and revert to 8B
Browse files
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-
|
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.
|
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 =
|
214 |
-
gr.Slider(0.0, 1.0, value = 0
|
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",
|