Spaces:
Sleeping
Sleeping
HanningChen
commited on
Commit
·
044bb04
1
Parent(s):
af40191
Fix bug
Browse files- 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}")
|