From 38d962cd5933d106e92f4e420cd617af0bc9cdb6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E7=8F=88=E7=BB=B4?= <591343188@qq.com>
Date: Tue, 14 Jan 2025 17:20:52 +0800
Subject: [PATCH] Add Qwen agent

---
 Qwen_agent.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/Qwen_agent.py b/Qwen_agent.py
index 2f491a9..7b168af 100644
--- a/Qwen_agent.py
+++ b/Qwen_agent.py
@@ -9,7 +9,6 @@ import requests
 # default: Load the model on the available device(s)
 model = Qwen2VLForConditionalGeneration.from_pretrained(
     "Qwen/Qwen2-VL-7B-Instruct",
-    # torch_dtype=torch.float16,
     torch_dtype="auto",
     device_map="auto"
 )
@@ -25,7 +24,6 @@ def qwen_prompt(img_list, messages):
     inputs = processor(
         text=[text],
         images=img_list,
-        # videos=video_inputs,
         padding=True,
         return_tensors="pt",
     )