Lemorra commited on
Commit
655d5ae
Β·
1 Parent(s): d79a053

πŸ› Fixed the bug for unexpected keyword

Browse files
Files changed (1) hide show
  1. src/utils/qwen_inference.py +1 -2
src/utils/qwen_inference.py CHANGED
@@ -78,7 +78,7 @@ class Qwen2_5(BaseModel):
78
  text = self.processor.apply_chat_template(
79
  messages, tokenize=False, add_generation_prompt=True
80
  )
81
- image_inputs, video_inputs, video_kwargs = process_vision_info(messages, return_video_kwargs=True)
82
  inputs = self.processor(
83
  text=[text],
84
  images=image_inputs,
@@ -86,7 +86,6 @@ class Qwen2_5(BaseModel):
86
  fps=1.0,
87
  padding=True,
88
  return_tensors="pt",
89
- **video_kwargs,
90
  )
91
  inputs = inputs.to("cuda")
92
  return inputs
 
78
  text = self.processor.apply_chat_template(
79
  messages, tokenize=False, add_generation_prompt=True
80
  )
81
+ image_inputs, video_inputs = process_vision_info(messages)
82
  inputs = self.processor(
83
  text=[text],
84
  images=image_inputs,
 
86
  fps=1.0,
87
  padding=True,
88
  return_tensors="pt",
 
89
  )
90
  inputs = inputs.to("cuda")
91
  return inputs