chore: update README.md models (#23)

Signed-off-by: Aaron Pham <contact@aarnphm.xyz>
bentoml · Sep 26, 2024 · 4aea71d · 4aea71d
1 parent 63a9c17
commit 4aea71d
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 43 deletions.
diff --git a/README.md b/README.md
diff --git a/gen_readme.py b/gen_readme.py
@@ -13,9 +13,7 @@
     "llama3": "Llama-3",
     "phi3": "Phi-3",
     "mistral": "Mistral",
-    "qwen2": "Qwen-2",
     "qwen2.5": "Qwen-2.5",
-    "gemma": "Gemma",
     "gemma2": "Gemma-2",
     "mixtral": "Mixtral",
     "mistral-large": "Mistral-Large",
@@ -40,6 +38,7 @@
 for yaml_file in yaml_files:
     with open(yaml_file, "r") as f:
         data = yaml.safe_load(f)
+        if data['name'] not in model_display_names: continue
         # Extract the HF model ID from routes.input.model.default
         for route in data.get("schema", {}).get("routes", []):
             for prop, details in route.get("input", {}).get("properties", {}).items():