Update mllama_train.py

Signed-off-by: Yu Yao <[email protected]>

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