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(