From a3b65be6b17922229ae690a866badcc2daa525d1 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:18:39 +0800 Subject: [PATCH] add Qwen interface --- Qwen_agent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Qwen_agent.py b/Qwen_agent.py index 6ebcc2e..cf765f9 100644 --- a/Qwen_agent.py +++ b/Qwen_agent.py @@ -1,10 +1,10 @@ from transformers import Qwen2VLForConditionalGeneration, AutoTokenizer, AutoProcessor from stream_pipeline import stream_pipeline from PIL import Image +from io import BytesIO import torch import ast import requests -from io import BytesIO # default: Load the model on the available device(s) model = Qwen2VLForConditionalGeneration.from_pretrained(