Add SOUP model option

app.py

@@ -35,7 +35,7 @@ os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
 
 ROOT_DIR = Path(__file__).resolve().parent
 CONFIG_PATH = ROOT_DIR / "configs" / "medical_vqa.yaml"
-VARIANT_ORDER = ["A1", "A2", "B1", "B2", "DPO", "PPO"]
+VARIANT_ORDER = ["A1", "A2", "B1", "B2", "DPO", "PPO", "SOUP"]
 MODEL_DISPLAY_NAMES = {
     "A1": "A1 LSTM",
     "A2": "A2 Transformer",
@@ -43,6 +43,7 @@ MODEL_DISPLAY_NAMES = {
     "B2": "B2 Fine-tuned",
     "DPO": "DPO Alignment",
     "PPO": "PPO RL refinement",
+    "SOUP": "SOUP Model Soup",
 }
 HF_MODEL_REPOS = {
     "A1": "SpringWang08/medical-vqa-a1",
@@ -51,6 +52,7 @@ HF_MODEL_REPOS = {
     "B2": "SpringWang08/medical-vqa-b2",
     "DPO": "SpringWang08/medical-vqa-dpo",
     "PPO": "SpringWang08/medical-vqa-ppo",
+    "SOUP": "SpringWang08/medical-vqa-soup",
 }
 
 with open(CONFIG_PATH, "r", encoding="utf-8") as f:
@@ -185,7 +187,7 @@ def _ensure_llava_bundle() -> dict[str, Any]:
         return llava_bundle
 
     wrapper, processor, base_model = _build_llava_base_and_processor()
-    adapter_variants = ["B2", "DPO", "PPO"]
+    adapter_variants = ["B2", "DPO", "PPO", "SOUP"]
     first_variant = adapter_variants[0]
     model = PeftModel.from_pretrained(
         base_model,