Skip to content

Commit b557cfd

Browse files
authored
Fix vlm engine changes in mcore (#15076)
1 parent 93738f3 commit b557cfd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

nemo/collections/vlm/inference/vlm_engine.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def generate(
3838
for i in range(len(prompts)):
3939
prompt = prompts[i]
4040
image = images[i] if images is not None else None
41-
prompt_tokens, image_dict = self.text_generation_controller.tokenize_prompt(prompt, image)
41+
prompt_tokens, image_dict = self.controller.tokenize_prompt(prompt, image)
4242

4343
# Reuse encoder_prompt from scheduler to pass image
4444
request_id = self.scheduler.add_request(

0 commit comments

Comments
 (0)