Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update
Browse files
app.py
CHANGED
@@ -88,13 +88,13 @@ MAX_MODEL_SIZE = ORIGINAL_DF["#Params (B)"].max()
|
|
88 |
|
89 |
def filter_models(
|
90 |
df: pd.DataFrame,
|
91 |
-
type_query: list,
|
92 |
-
size_query: list,
|
93 |
-
precision_query: list,
|
94 |
-
add_special_tokens_query: list,
|
95 |
-
num_few_shots_query: list,
|
96 |
-
version_query: list,
|
97 |
-
vllm_query: list,
|
98 |
) -> pd.DataFrame:
|
99 |
print(f"Initial df shape: {df.shape}")
|
100 |
print(f"Initial df content:\n{df}")
|
|
|
88 |
|
89 |
def filter_models(
|
90 |
df: pd.DataFrame,
|
91 |
+
type_query: list[str],
|
92 |
+
size_query: list[str],
|
93 |
+
precision_query: list[str],
|
94 |
+
add_special_tokens_query: list[str],
|
95 |
+
num_few_shots_query: list[str],
|
96 |
+
version_query: list[str],
|
97 |
+
vllm_query: list[str],
|
98 |
) -> pd.DataFrame:
|
99 |
print(f"Initial df shape: {df.shape}")
|
100 |
print(f"Initial df content:\n{df}")
|