Fix search bar by not filtered models with unknown model type

app.py

@@ -225,6 +225,7 @@ def update_table(hidden_df: pd.DataFrame, current_columns_df: pd.DataFrame, colu
     return df
 
 def search_table(df: pd.DataFrame, query: str) -> pd.DataFrame:
+    print(query)
     return df[(df[AutoEvalColumn.dummy.name].str.contains(query, case=False))]
 
 def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
@@ -258,6 +259,7 @@ def filter_models(
         filtered_df = df[df[AutoEvalColumn.still_on_hub.name] == True]
 
     type_emoji = [t[0] for t in type_query]
+    print(type_emoji)
     filtered_df = filtered_df[df[AutoEvalColumn.model_type_symbol.name].isin(type_emoji)]
     filtered_df = filtered_df[df[AutoEvalColumn.precision.name].isin(precision_query)]
 
@@ -325,12 +327,14 @@ with demo:
                                 ModelType.FT.to_str(),
                                 ModelType.IFT.to_str(),
                                 ModelType.RL.to_str(),
+                                ModelType.Unknown.to_str(),
                             ],
                             value=[
                                 ModelType.PT.to_str(),
                                 ModelType.FT.to_str(),
                                 ModelType.IFT.to_str(),
                                 ModelType.RL.to_str(),
+                                ModelType.Unknown.to_str(),
                             ],
                             interactive=True,
                             elem_id="filter-columns-type",

src/display_models/get_model_metadata.py

@@ -85,7 +85,7 @@ def get_model_size(model_name, model_info):
             with init_empty_weights():
                 model = AutoModel.from_config(config, trust_remote_code=False)
             return round(sum(p.numel() for p in model.parameters() if p.requires_grad) / 1e9, 3)
-        except (EnvironmentError, ValueError): # model config not found, likely private  
+        except (EnvironmentError, ValueError, KeyError): # model config not found, likely private  
             try:
                 size_match = re.search(size_pattern, model_name.lower())
                 size = size_match.group(0)

src/display_models/model_metadata_type.py

@@ -14,7 +14,7 @@ class ModelType(Enum):
     FT = ModelInfo(name="fine-tuned", symbol="🔶")
     IFT = ModelInfo(name="instruction-tuned", symbol="⭕")
     RL = ModelInfo(name="RL-tuned", symbol="🟦")
-    Unknown = ModelInfo(name="Unknown, add type to request file!", symbol="?")
+    Unknown = ModelInfo(name="Unknown", symbol="?")
 
     def to_str(self, separator=" "):
         return f"{self.value.symbol}{separator}{self.value.name}"