Fix CID runtime step limit
Browse files- README.md +1 -1
- app.py +4 -3
- requirements.txt +1 -1
README.md
CHANGED
|
@@ -24,4 +24,4 @@ Interactive Hugging Face Space for the 398.8M-parameter `fwerkor/CID-v1-0.4B` ch
|
|
| 24 |
|
| 25 |
The demo exposes CID's revisable Display and runtime trace, with verified example prompts for the released 0.4B checkpoint. Users may optionally provide a task-local document workspace; CID can then issue read-only `workspace_search` and `workspace_read` information needs against those documents while reasoning.
|
| 26 |
|
| 27 |
-
The model is mechanism-focused rather than a general-purpose chat model. The Space uses the published checkpoint and pins the CID runtime implementation to commit `
|
|
|
|
| 24 |
|
| 25 |
The demo exposes CID's revisable Display and runtime trace, with verified example prompts for the released 0.4B checkpoint. Users may optionally provide a task-local document workspace; CID can then issue read-only `workspace_search` and `workspace_read` information needs against those documents while reasoning.
|
| 26 |
|
| 27 |
+
The model is mechanism-focused rather than a general-purpose chat model. The Space uses the published checkpoint and pins the CID runtime implementation to commit `dffbec320dfd0ae264a8d0e86946d29dfe704843`.
|
app.py
CHANGED
|
@@ -33,7 +33,7 @@ from cid.runtime.engine import RuntimeConfig
|
|
| 33 |
MODEL_ID = "fwerkor/CID-v1-0.4B"
|
| 34 |
MODEL_REVISION = "b291b799cf654f7074e6bebb48d49fee31c17824"
|
| 35 |
MODEL_CHOICES = [("CID-v1-0.4B · 398.8M", MODEL_ID)]
|
| 36 |
-
DEMO_CODE_REVISION = "
|
| 37 |
MAX_PROMPT_CHARS = 4_000
|
| 38 |
MAX_WORKSPACE_CHARS = 24_000
|
| 39 |
MAX_DOCUMENTS = 12
|
|
@@ -450,7 +450,8 @@ def run_cid(
|
|
| 450 |
allocation_threshold=float(allocation_threshold),
|
| 451 |
)
|
| 452 |
runtime_config = RuntimeConfig(
|
| 453 |
-
max_steps=
|
|
|
|
| 454 |
max_wall_time_s=55.0,
|
| 455 |
binding_threshold=float(binding_threshold),
|
| 456 |
trace_display=True,
|
|
@@ -612,7 +613,7 @@ with gr.Blocks(
|
|
| 612 |
)
|
| 613 |
|
| 614 |
with gr.Accordion("Runtime controls", open=False):
|
| 615 |
-
max_steps = gr.Slider(
|
| 616 |
with gr.Row():
|
| 617 |
need_threshold = gr.Slider(
|
| 618 |
0.05,
|
|
|
|
| 33 |
MODEL_ID = "fwerkor/CID-v1-0.4B"
|
| 34 |
MODEL_REVISION = "b291b799cf654f7074e6bebb48d49fee31c17824"
|
| 35 |
MODEL_CHOICES = [("CID-v1-0.4B · 398.8M", MODEL_ID)]
|
| 36 |
+
DEMO_CODE_REVISION = "dffbec3"
|
| 37 |
MAX_PROMPT_CHARS = 4_000
|
| 38 |
MAX_WORKSPACE_CHARS = 24_000
|
| 39 |
MAX_DOCUMENTS = 12
|
|
|
|
| 450 |
allocation_threshold=float(allocation_threshold),
|
| 451 |
)
|
| 452 |
runtime_config = RuntimeConfig(
|
| 453 |
+
max_steps=256,
|
| 454 |
+
max_total_steps=int(max_steps),
|
| 455 |
max_wall_time_s=55.0,
|
| 456 |
binding_threshold=float(binding_threshold),
|
| 457 |
trace_display=True,
|
|
|
|
| 613 |
)
|
| 614 |
|
| 615 |
with gr.Accordion("Runtime controls", open=False):
|
| 616 |
+
max_steps = gr.Slider(1, 256, value=128, step=1, label="Max CID model steps")
|
| 617 |
with gr.Row():
|
| 618 |
need_threshold = gr.Slider(
|
| 619 |
0.05,
|
requirements.txt
CHANGED
|
@@ -2,4 +2,4 @@ spaces
|
|
| 2 |
transformers==4.57.1
|
| 3 |
safetensors>=0.4.5
|
| 4 |
accelerate>=1.2
|
| 5 |
-
git+https://github.com/fwerkor/continuous-interaction-diffusion.git@
|
|
|
|
| 2 |
transformers==4.57.1
|
| 3 |
safetensors>=0.4.5
|
| 4 |
accelerate>=1.2
|
| 5 |
+
git+https://github.com/fwerkor/continuous-interaction-diffusion.git@dffbec320dfd0ae264a8d0e86946d29dfe704843
|