bkhmsi commited on
Commit
5c65b5e
·
1 Parent(s): 3768f67

fixed config path

Browse files
Files changed (1) hide show
  1. 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