We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 93738f3 commit b557cfdCopy full SHA for b557cfd
nemo/collections/vlm/inference/vlm_engine.py
@@ -38,7 +38,7 @@ def generate(
38
for i in range(len(prompts)):
39
prompt = prompts[i]
40
image = images[i] if images is not None else None
41
- prompt_tokens, image_dict = self.text_generation_controller.tokenize_prompt(prompt, image)
+ prompt_tokens, image_dict = self.controller.tokenize_prompt(prompt, image)
42
43
# Reuse encoder_prompt from scheduler to pass image
44
request_id = self.scheduler.add_request(
0 commit comments