fwerkor commited on
Commit
42adf6c
·
1 Parent(s): 81f1130

Fix CID runtime step limit

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +4 -3
  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 `6fab022722701e51e742de6a135bbf1e15bfd7e0`.
 
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 = "6fab022"
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=int(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(8, 64, value=32, step=4, label="Max CID model steps")
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@6fab022722701e51e742de6a135bbf1e15bfd7e0
 
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