Rename Mochi integration test correctly (#10220)

rename integration test
huggingface · Dec 18, 2024 · f66bd32 · f66bd32
1 parent c4c99c3
commit f66bd32
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/pipelines/mochi/test_mochi.py b/tests/pipelines/mochi/test_mochi.py
@@ -275,7 +275,7 @@ def tearDown(self):
         gc.collect()
         torch.cuda.empty_cache()
 
-    def test_cogvideox(self):
+    def test_mochi(self):
         generator = torch.Generator("cpu").manual_seed(0)
 
         pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview", torch_dtype=torch.float16)