IlyasMoutawwakil's picture
update
3d7033f
raw
history blame
No virus
346 Bytes
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())