SPB Git

spb/wp33_uqo

Public MIT
489 B · 16 lines python
1import os, glob, pytest23def _model_dir():4    env = os.environ.get("TITAN_MODEL_DIR")5    if env:6        return env7    hits = glob.glob(os.path.expanduser("~/.cache/huggingface/hub/models--mlx-community--OLMoE-1B-7B-0125-Instruct-4bit/snapshots/*"))8    return hits[0] if hits else None910@pytest.fixture(scope="session")11def model_dir():12    d = _model_dir()13    if d is None:14        pytest.skip("OLMoE-1B-7B-0125-Instruct-4bit not in the HF cache (set TITAN_MODEL_DIR)")15    return d16
Files · 1df8d30cf1031f106c8a51dd4503d5c1aa1d099e
Menu

Theme