eduagarcia commited on
Commit
ab44cd6
1 Parent(s): 09cd30b

filter quantized models from the collection

Browse files
Files changed (1) hide show
  1. 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 = []