return (fastdeploy_path, huggingface_path)
(base_model_name: str)
| 201 | |
| 202 | |
| 203 | def get_torch_model_path(base_model_name: str) -> str: |
| 204 | """return (fastdeploy_path, huggingface_path)""" |
| 205 | # FastDeploy model path |
| 206 | fd_base_path = os.getenv("MODEL_PATH") |
| 207 | # HuggingFace model path |
| 208 | torch_model_path = os.path.join( |
| 209 | fd_base_path, |
| 210 | "torch", |
| 211 | base_model_name, |
| 212 | ) |
| 213 | |
| 214 | return torch_model_path |
| 215 | |
| 216 | |
| 217 | def check_tokens_id_and_text_close( |