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", )