fix path

huggingface · Sep 25, 2024 · a17100c · a17100c
1 parent 3d8405d
commit a17100c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/mllama/test_processor_mllama.py b/tests/models/mllama/test_processor_mllama.py
@@ -30,7 +30,7 @@
 @require_vision
 class MllamaProcessorTest(unittest.TestCase):
  def setUp(self):
- self.checkpoint = "Llama-3.2-11B-Vision-Instruct" # TODO: change
+ self.checkpoint = "hf-internal-testing/mllama-11b" # TODO: change
  self.processor = MllamaProcessor.from_pretrained(self.checkpoint)
  self.image1 = Image.new("RGB", (224, 220))
  self.image2 = Image.new("RGB", (512, 128))