lichen/detecttracking
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add Qwen interface

Browse Source
This commit is contained in:
刘珈维
2025-01-14 17:18:39 +08:00
parent 6ede9580cb
commit a3b65be6b1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Qwen_agent.py
View File

@ -1,10 +1,10 @@
from transformers import Qwen2VLForConditionalGeneration, AutoTokenizer, AutoProcessor from transformers import Qwen2VLForConditionalGeneration, AutoTokenizer, AutoProcessor
from stream_pipeline import stream_pipeline from stream_pipeline import stream_pipeline
from PIL import Image from PIL import Image
from io import BytesIO
import torch import torch
import ast import ast
import requests import requests
from io import BytesIO
# default: Load the model on the available device(s) # default: Load the model on the available device(s)
model = Qwen2VLForConditionalGeneration.from_pretrained( model = Qwen2VLForConditionalGeneration.from_pretrained(