Rename Mochi integration test correctly (#10220)

rename integration test

Rename Mochi integration test correctly (#10220)
rename integration test
f66bd326 · Aryan · GitHub · c4c99c39 · f66bd326
Unverified Commit f66bd326 authored Dec 18, 2024 by Aryan Committed by GitHub Dec 18, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/pipelines/mochi/test_mochi.py tests/pipelines/mochi/test_mochi.py +1 -1

No files found.
--- a/tests/pipelines/mochi/test_mochi.py
+++ b/tests/pipelines/mochi/test_mochi.py
@@ -275,7 +275,7 @@ class MochiPipelineIntegrationTests(unittest.TestCase):
        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)