File size: 384 Bytes
80dc74c
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from huggingface_hub import hf_hub_download

supported_models_path = hf_hub_download(
    repo_id="jacopoteneggi/IBYDMT",
    filename="supported_models.txt",
    repo_type="dataset",
)

SUPPORTED_MODELS = []
with open(supported_models_path, "r") as f:
    for line in f:
        line = line.strip()
        model_name, _ = line.split(",")
        SUPPORTED_MODELS.append(model_name)