File size: 288 Bytes
db3982a
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from __future__ import annotations
import os
from pathlib import Path

ROOT = Path(os.environ.get("RIEMANN_ROOT", "C:/RiemannModel"))
DATA_DIR = ROOT / "data"
CHECKPOINT_ROOT = ROOT / "checkpoints"
CONTROLLER_ROOT = ROOT / "ntk_controllers"
HF_CACHE_DIR = str(ROOT / "hf_cache")