Update gradio_app/config.py

gradio_app/config.py

@@ -5,56 +5,25 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # LoRA configurations
-# LORA_CONFIGS = {
-#     "Gemma-3-1B-Instruct-Vi-Medical-LoRA": {
-#         "base_model": "unsloth/gemma-3-1b-it",
-#         "lora_adapter": "danhtran2mind/Gemma-3-1B-Instruct-Vi-Medical-LoRA"
-#     },
-#     "Gemma-3-1B-GRPO-Vi-Medical-LoRA": {
-#         "base_model": "unsloth/gemma-3-1b-it",
-#         "lora_adapter": "danhtran2mind/Gemma-3-1B-GRPO-Vi-Medical-LoRA"
-#     },
-#     "Llama-3.2-3B-Instruct-Vi-Medical-LoRA": {
-#         "base_model": "unsloth/Llama-3.2-3B-Instruct",
-#         "lora_adapter": "danhtran2mind/Llama-3.2-3B-Instruct-Vi-Medical-LoRA"
-#     },
-#     "Llama-3.2-1B-Instruct-Vi-Medical-LoRA": {
-#         "base_model": "unsloth/Llama-3.2-1B-Instruct",
-#         "lora_adapter": "danhtran2mind/Llama-3.2-1B-Instruct-Vi-Medical-LoRA"
-#     },
-#     "Llama-3.2-3B-Reasoning-Vi-Medical-LoRA": {
-#         "base_model": "unsloth/Llama-3.2-3B-Instruct",
-#         "lora_adapter": "danhtran2mind/Llama-3.2-3B-Reasoning-Vi-Medical-LoRA"
-#     },
-#     "Qwen-3-0.6B-Instruct-Vi-Medical-LoRA": {
-#         "base_model": "Qwen/Qwen3-0.6B",
-#         "lora_adapter": "danhtran2mind/Qwen-3-0.6B-Instruct-Vi-Medical-LoRA"
-#     },
-#     "Qwen-3-0.6B-Reasoning-Vi-Medical-LoRA": {
-#         "base_model": "Qwen/Qwen3-0.6B",
-#         "lora_adapter": "danhtran2mind/Qwen-3-0.6B-Reasoning-Vi-Medical-LoRA"
-#     }
-# }
-
 LORA_CONFIGS = {
     "Gemma-3-1B-Instruct-Vi-Medical-LoRA": {
-        "base_model": "google/gemma-3-1b-it",
+        "base_model": "unsloth/gemma-3-1b-it",
         "lora_adapter": "danhtran2mind/Gemma-3-1B-Instruct-Vi-Medical-LoRA"
     },
     "Gemma-3-1B-GRPO-Vi-Medical-LoRA": {
-        "base_model": "google/gemma-3-1b-it",
+        "base_model": "unsloth/gemma-3-1b-it",
         "lora_adapter": "danhtran2mind/Gemma-3-1B-GRPO-Vi-Medical-LoRA"
     },
     "Llama-3.2-3B-Instruct-Vi-Medical-LoRA": {
-        "base_model": "meta-llama/Llama-3.2-3B-Instruct",
+        "base_model": "unsloth/Llama-3.2-3B-Instruct",
         "lora_adapter": "danhtran2mind/Llama-3.2-3B-Instruct-Vi-Medical-LoRA"
     },
     "Llama-3.2-1B-Instruct-Vi-Medical-LoRA": {
-        "base_model": "meta-llama/Llama-3.2-1B-Instruct",
+        "base_model": "unsloth/Llama-3.2-1B-Instruct",
         "lora_adapter": "danhtran2mind/Llama-3.2-1B-Instruct-Vi-Medical-LoRA"
     },
     "Llama-3.2-3B-Reasoning-Vi-Medical-LoRA": {
-        "base_model": "meta-llama/Llama-3.2-3B-Instruct",
+        "base_model": "unsloth/Llama-3.2-3B-Instruct",
         "lora_adapter": "danhtran2mind/Llama-3.2-3B-Reasoning-Vi-Medical-LoRA"
     },
     "Qwen-3-0.6B-Instruct-Vi-Medical-LoRA": {
@@ -67,6 +36,37 @@ LORA_CONFIGS = {
     }
 }
 
+# LORA_CONFIGS = {
+#     "Gemma-3-1B-Instruct-Vi-Medical-LoRA": {
+#         "base_model": "google/gemma-3-1b-it",
+#         "lora_adapter": "danhtran2mind/Gemma-3-1B-Instruct-Vi-Medical-LoRA"
+#     },
+#     "Gemma-3-1B-GRPO-Vi-Medical-LoRA": {
+#         "base_model": "google/gemma-3-1b-it",
+#         "lora_adapter": "danhtran2mind/Gemma-3-1B-GRPO-Vi-Medical-LoRA"
+#     },
+#     "Llama-3.2-3B-Instruct-Vi-Medical-LoRA": {
+#         "base_model": "meta-llama/Llama-3.2-3B-Instruct",
+#         "lora_adapter": "danhtran2mind/Llama-3.2-3B-Instruct-Vi-Medical-LoRA"
+#     },
+#     "Llama-3.2-1B-Instruct-Vi-Medical-LoRA": {
+#         "base_model": "meta-llama/Llama-3.2-1B-Instruct",
+#         "lora_adapter": "danhtran2mind/Llama-3.2-1B-Instruct-Vi-Medical-LoRA"
+#     },
+#     "Llama-3.2-3B-Reasoning-Vi-Medical-LoRA": {
+#         "base_model": "meta-llama/Llama-3.2-3B-Instruct",
+#         "lora_adapter": "danhtran2mind/Llama-3.2-3B-Reasoning-Vi-Medical-LoRA"
+#     },
+#     "Qwen-3-0.6B-Instruct-Vi-Medical-LoRA": {
+#         "base_model": "Qwen/Qwen3-0.6B",
+#         "lora_adapter": "danhtran2mind/Qwen-3-0.6B-Instruct-Vi-Medical-LoRA"
+#     },
+#     "Qwen-3-0.6B-Reasoning-Vi-Medical-LoRA": {
+#         "base_model": "Qwen/Qwen3-0.6B",
+#         "lora_adapter": "danhtran2mind/Qwen-3-0.6B-Reasoning-Vi-Medical-LoRA"
+#     }
+# }
+
 # Model settings
 MAX_INPUT_TOKEN_LENGTH = 4096
 DEFAULT_MAX_NEW_TOKENS = 512