update

models.py

@@ -15,6 +15,18 @@ RECOMMENDED_MODELS = {
         "description": "3B parameter vision-language model with instruction tuning",
         "dtype": "bfloat16",
         "device": "cuda"
+    },
+    "Qwen2-VL-7B": {
+        "id": "Qwen/Qwen2-VL-7B",
+        "description": "7B parameter vision-language model (Qwen2 series)",
+        "dtype": "bfloat16",
+        "device": "cuda"
+    },
+    "Qwen2-VL-7B-Instruct": {
+        "id": "Qwen/Qwen2-VL-7B-Instruct",
+        "description": "7B parameter vision-language model with instruction tuning (Qwen2 series)",
+        "dtype": "bfloat16",
+        "device": "cuda"
     }
 }
 
@@ -32,6 +44,47 @@ DEFAULT_GENERATION_PARAMS = {
     "no_repeat_ngram_size": 0
 }
 
+# Generation parameter presets for quick selection
+GENERATION_PRESETS = {
+    "Default": DEFAULT_GENERATION_PARAMS,
+    "Creative": {
+        "max_new_tokens": 1024,
+        "temperature": 0.9,
+        "top_p": 0.95,
+        "top_k": 60,
+        "repetition_penalty": 1.1,
+        "do_sample": True,
+        "num_beams": 1,
+        "early_stopping": True,
+        "length_penalty": 1.0,
+        "no_repeat_ngram_size": 0
+    },
+    "Precise": {
+        "max_new_tokens": 1024,
+        "temperature": 0.5,
+        "top_p": 0.85,
+        "top_k": 40,
+        "repetition_penalty": 1.2,
+        "do_sample": True,
+        "num_beams": 3,
+        "early_stopping": True,
+        "length_penalty": 1.0,
+        "no_repeat_ngram_size": 3
+    },
+    "Deterministic": {
+        "max_new_tokens": 1024,
+        "temperature": 0.0,
+        "top_p": 1.0,
+        "top_k": 50,
+        "repetition_penalty": 1.0,
+        "do_sample": False,
+        "num_beams": 5,
+        "early_stopping": True,
+        "length_penalty": 1.0,
+        "no_repeat_ngram_size": 0
+    }
+}
+
 def get_model_info(model_name):
     """
     Returns the model information for a given model name.