From 16a6fc6458fb821ae33ddd044631f53b36ff1480 Mon Sep 17 00:00:00 2001
From: Yu Yao <54727607+yaoyu-33@users.noreply.github.com>
Date: Fri, 22 Nov 2024 11:31:05 -0800
Subject: [PATCH] Update mllama_train.py

Signed-off-by: Yu Yao <54727607+yaoyu-33@users.noreply.github.com>
---
 tests/collections/vlm/mllama_train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/collections/vlm/mllama_train.py b/tests/collections/vlm/mllama_train.py
index 3e37c41a4426..b347fcc3dc91 100644
--- a/tests/collections/vlm/mllama_train.py
+++ b/tests/collections/vlm/mllama_train.py
@@ -23,7 +23,7 @@
 from transformers import AutoProcessor
 
 from nemo import lightning as nl
-from nemo.collections import llm, vlm
+from nemo.collections import vlm
 from nemo.collections.common.tokenizers.huggingface.auto_tokenizer import AutoTokenizer
 from nemo.collections.llm.api import train
 from nemo.lightning import AutoResume, NeMoLogger