trllm / models.py
aka7774's picture
Upload 8 files
581f159 verified
raw
history blame
362 Bytes
import json
def load():
with open('models.json', 'r', encoding='utf-8') as f:
models = json.load(f)
return models
def get_examples():
examples = []
for model in models.keys():
examples.append(list(models[model].values()))
return examples
def get_head_options():
return models[list(models.keys())[0]]
models = load()