File size: 346 Bytes
3d7033f
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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())