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 · 7ab0c879e1457dd71afdcec6e1bbfae1fc1ef904
Menu

Theme