Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
eduagarcia
commited on
Commit
•
ab44cd6
1
Parent(s):
09cd30b
filter quantized models from the collection
Browse files- src/tools/collections.py +3 -0
src/tools/collections.py
CHANGED
@@ -43,6 +43,9 @@ def update_collections(df: DataFrame):
|
|
43 |
except HfHubHTTPError:
|
44 |
continue
|
45 |
|
|
|
|
|
|
|
46 |
ix = 0
|
47 |
for size in intervals:
|
48 |
interval_scores = []
|
|
|
43 |
except HfHubHTTPError:
|
44 |
continue
|
45 |
|
46 |
+
#filter quantized models
|
47 |
+
df = df[df[AutoEvalColumn.precision.name].isin(['bfloat16', 'float16'])]
|
48 |
+
|
49 |
ix = 0
|
50 |
for size in intervals:
|
51 |
interval_scores = []
|