Merge pull request #44 from Supahands/chore-add_supa_ollama_models

Chore add supa ollama models

Author: OriginalByteMe

hugging_face_to_guff.py

@@ -85,7 +85,7 @@ def download_model(
         private: bool = False,
         ollama_upload: bool = False,
         hf_upload: bool = False,
-        clean_run: bool = False,  # New parameter
+        clean_run: bool = False,  
     ):
         logger.info(f"Downloading model {model_id}...")
         import subprocess
@@ -159,8 +159,8 @@ def convert_to_gguf(
         branch: str = "",
         filter_path: str = "",
         ollama_upload: bool = False,
-        hf_upload: bool = False,  # New parameter
-        clean_run: bool = False,  # New parameter
+        hf_upload: bool = False,  
+        clean_run: bool = False,  
     ):
         """Convert model to GGUF format with multiple quantization types and push to Ollama"""
         logger.info(f"Converting model with quantization types: {quanttypes}")
@@ -246,7 +246,7 @@ def push_to_ollama(
         modelname: str,
         source_model_id: str,
         username: str,
-        clean_run: bool = False,  # New parameter
+        clean_run: bool = False,  
     ):
         """Push converted models to Ollama using tags for different quantizations"""
         logger.info("Pushing models to Ollama...")
@@ -375,7 +375,7 @@ def main(
     private: bool = False,
     ollama_upload: bool = False,
     hf_upload: bool = False,
-    clean_run: bool = False,  # New parameter
+    clean_run: bool = False,  
 ):
     logger.info(f"Starting conversion process for {modelowner}/{modelname}")
     converter = ModelConverter()

ollama_service.py

@@ -15,14 +15,18 @@
 # Default server port.
 MODEL_IDS: list[str] = [
     "llama3",
+    "llama3.1",
     "llama3.2",
     "mistral",
     "gemma2",
     "qwen2.5",
+    "yi",
     "aisingapore/gemma2-9b-cpt-sea-lionv3-instruct",
     "hf.co/Supa-AI/malaysian-Llama-3.2-3B-Instruct-Q8_0-GGUF",
-    "hf.co/Supa-AI/gemma2-9b-cpt-sahabatai-v1-instruct-q8_0-gguf",
-    "hf.co/Supa-AI/llama3-8b-cpt-sahabatai-v1-instruct-q8_0-gguf"
+    "Supa-AI/gemma2-9b-cpt-sahabatai-v1-instruct:q8_0",
+    "Supa-AI/llama3-8b-cpt-sahabatai-v1-instruct:q8_0",
+    "Supa-AI/gemma2-9b-cpt-sahabatai-v1-base:q8_0",
+    "Supa-AI/ministral-8b-instruct-2410:q8_0"
 ]
 
 OLLAMA_PORT: int = 11434