Update stage_1/model_setup.py

stage_1/model_setup.py

@@ -23,7 +23,8 @@ config = {
 }
 
 # Model and Device Configuration
-MODEL_ID = "qwen2.5-3B-instruct-ft/latest"
+# Model setup
+MODEL_ID = "Qwen/Qwen2.5-VL-3B-Instruct"
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Image Configuration