(base_model_name: str)
| 192 | |
| 193 | |
| 194 | def get_paddle_model_path(base_model_name: str) -> str: |
| 195 | fd_base_path = os.getenv("MODEL_PATH") |
| 196 | if fd_base_path: |
| 197 | fd_model_path = os.path.join(fd_base_path, base_model_name) |
| 198 | else: |
| 199 | fd_model_path = base_model_name |
| 200 | return fd_model_path |
| 201 | |
| 202 | |
| 203 | def get_torch_model_path(base_model_name: str) -> str: |