fixing function to load

main.py

@@ -42,9 +42,10 @@ max_pixels = 1280 * 28 * 28
 processor = AutoProcessor.from_pretrained(
     checkpoint, min_pixels=min_pixels, max_pixels=max_pixels
 )
-model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
+model = Qwen2VLForConditionalGeneration.from_pretrained(
     checkpoint,
-    torch_dtype=torch.bfloat16,
+    # torch_dtype=torch.bfloat16,
+    torch_dtype="auto",
     device_map="auto",
     # attn_implementation="flash_attention_2",
 )