Skip to content

Commit e55a3c9

Browse files
committed
feat: reduce block sizes and add tearDownClass for cleanup in LTX2 smoke test
1 parent f164229 commit e55a3c9

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

src/maxdiffusion/tests/generate_ltx2_smoke_test.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,13 @@ def test_ltx2_inference(self):
9797
print(f"Audio produced with shape: {audios[0].shape}")
9898
self.assertGreater(len(audios), 0)
9999

100+
@classmethod
101+
def tearDownClass(cls):
102+
del cls.pipeline
103+
import gc
104+
105+
gc.collect()
106+
100107

101108
if __name__ == "__main__":
102109
unittest.main()

0 commit comments

Comments
 (0)