HanningChen commited on
Commit
044bb04
·
1 Parent(s): af40191
Files changed (1) hide show
  1. webui/runner.py +4 -0
webui/runner.py CHANGED
@@ -401,6 +401,10 @@ class ModelRunner:
401
  hdv_dim: int = 256,
402
  taskclip_ckpt: str = "./test_model/default/decoder.pt",
403
  viz_mode: str = "bbox",
 
 
 
 
404
  ) -> Dict[str, Any]:
405
  if vlm_model not in ["imagebind", "vit-b", "vit-l"]:
406
  raise ValueError(f"Unknown vlm_model: {vlm_model}")
 
401
  hdv_dim: int = 256,
402
  taskclip_ckpt: str = "./test_model/default/decoder.pt",
403
  viz_mode: str = "bbox",
404
+ hw_noise_dist: str = "none",
405
+ hw_noise_width: int = 0,
406
+ hw_noise_strength: int = 0,
407
+ hdc_bits: int = 32
408
  ) -> Dict[str, Any]:
409
  if vlm_model not in ["imagebind", "vit-b", "vit-l"]:
410
  raise ValueError(f"Unknown vlm_model: {vlm_model}")