Spaces:
Running
Running
fixed config path
Browse files- router_backend.py +1 -1
router_backend.py
CHANGED
|
@@ -195,7 +195,7 @@ def build_model(model_id: str, hf_token: str, use_cache: bool = True):
|
|
| 195 |
|
| 196 |
parent_path = pathlib.Path(__file__).parent
|
| 197 |
|
| 198 |
-
model_config.config_path = f"{parent_path}/configs/{model_id}.yml"
|
| 199 |
|
| 200 |
model_config.torch_dtype = torch.bfloat16
|
| 201 |
model_config.use_bfloat16 = True
|
|
|
|
| 195 |
|
| 196 |
parent_path = pathlib.Path(__file__).parent
|
| 197 |
|
| 198 |
+
model_config.config_path = f"{parent_path}/configs/{model_id.replace('-', '_')}.yml"
|
| 199 |
|
| 200 |
model_config.torch_dtype = torch.bfloat16
|
| 201 |
model_config.use_bfloat16 = True
|