from huggingface_hub import hf_hub_download
import pandas as pd


hf_hub_download(
    repo_id="optimum/llm-perf-dataset",
    filename="open-llm.csv",
    local_dir="dataset",
    repo_type="dataset",
)

open_llm = pd.read_csv("dataset/open-llm.csv")
print(open_llm["Arch"].unique())
print(open_llm[open_llm["Arch"] == "rwkv"]["Model"].unique())