| 47 | |
| 48 | |
| 49 | def llm_models_root(check: bool = False) -> Optional[Path]: |
| 50 | root = Path("/home/scratch.trt_llm_data_ci/llm-models/") |
| 51 | |
| 52 | if "LLM_MODELS_ROOT" in os.environ: |
| 53 | root = Path(os.environ.get("LLM_MODELS_ROOT")) |
| 54 | |
| 55 | if not root.exists(): |
| 56 | root = Path("/scratch.trt_llm_data/llm-models/") |
| 57 | |
| 58 | if check: |
| 59 | assert root.exists(), ( |
| 60 | "You must set LLM_MODELS_ROOT env or be able to access /home/scratch.trt_llm_data_ci to run this test" |
| 61 | ) |
| 62 | |
| 63 | return root if root.exists() else None |
| 64 | |
| 65 | |
| 66 | def llm_datasets_root() -> str: |