Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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 = []
|