Test local import

Author: Jack-Khuu

torchchat/model.py

@@ -29,11 +29,7 @@
     RowwiseParallel,
     SequenceParallel,
 )
-# TODO: remove this after we figure out where in torchtune an `evaluate` module
-# is being imported, which is being confused with huggingface's `evaluate``.
-import lm_eval  # noqa 
 from torch.nn import functional as F
-from torchtune.models.clip import clip_vision_encoder
 from torchtune.models.llama3_1._component_builders import llama3_1 as llama3_1_builder
 from torchtune.models.llama3_2_vision._component_builders import (
     llama3_2_vision_decoder,
@@ -227,6 +223,9 @@ def _flamingo(cls):
 
     @classmethod
     def _llava(cls):
+        # TODO: Import locally until we find where in torchtune an `evaluate` module
+        # is being imported, which is being confused with huggingface's `evaluate``.
+        from torchtune.models.clip import clip_vision_encoder
         return cls(
             model_type=ModelType.Llava,
             modules={