Add Qwen agent
This commit is contained in:
@ -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",
|
||||
)
|
||||
|
Reference in New Issue
Block a user