Teraflops / src /utils.py
Rajesh0279's picture
Upload 3 files
48a990e verified
raw
history blame
762 Bytes
# utils.py
def get_all_models_from_database(db):
models = []
for tier, categories in db.items():
for category, items in categories.items():
for model in items:
display_name = f"{model['name']} | {model['size']} | {category} | {tier}"
models.append({
"display": display_name,
"name": model["name"],
"size": model["size"],
"description": model["description"],
"cost_a100": model.get("cost(A100)", "₹0"),
"cost_h100": model.get("cost(H100)", "₹0"),
"tier": tier,
"category": category
})
return models