File size: 531 Bytes
9118991
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
"""Storage resolution only; independent of the other quantization methods."""
from pathlib import Path
from huggingface_hub.constants import HF_HUB_CACHE

REVISIONS = {
    "ouro": ("ByteDance/Ouro-1.4B", "e3b1e0993b1231a51d6069a870476dda4162c00a"),
    "huginn": ("tomg-group-umd/huginn-0125", "bb6621b65e90b6a4b9b29ef88dc83866d450470c"),
}


def pinned_snapshot(model: str) -> Path:
    repository, revision = REVISIONS[model]
    return Path(HF_HUB_CACHE) / ("models--" + repository.replace("/", "--")) / "snapshots" / revision