We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a764e8d commit d45d5b6Copy full SHA for d45d5b6
1 file changed
src/maxdiffusion/tests/ltx2_parity_test.py
@@ -11,6 +11,7 @@
11
from maxdiffusion import pyconfig
12
from maxdiffusion.max_utils import create_device_mesh
13
from maxdiffusion.models.ltx2.transformer_ltx2 import LTX2VideoTransformerBlock, LTX2VideoTransformer3DModel
14
+import torch
15
16
17
class LTX2TransformerParityTest(unittest.TestCase):
0 commit comments