From 508eaa22df68649a5969eba1b12ef2fc69d70c64 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Thu, 24 Oct 2024 00:34:50 +0300
Subject: [PATCH] Update t2v_synth_mochi.py

---
 mochi_preview/t2v_synth_mochi.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/mochi_preview/t2v_synth_mochi.py b/mochi_preview/t2v_synth_mochi.py
index 5cd8f01..6507700 100644
--- a/mochi_preview/t2v_synth_mochi.py
+++ b/mochi_preview/t2v_synth_mochi.py
@@ -23,6 +23,8 @@ except:
     is_accelerate_available = False
     pass
 
+from .dit.joint_model.asymm_models_joint import AsymmDiTJoint
+
 MAX_T5_TOKEN_LENGTH = 256
 
 def unnormalize_latents(
@@ -125,9 +127,7 @@ class T2VSynthMochiModel:
         self.offload_device = offload_device
 
         with t("construct_dit"):
-                from .dit.joint_model.asymm_models_joint import (
-                    AsymmDiTJoint,
-                )
+                
                 model: nn.Module = torch.nn.utils.skip_init(
                     AsymmDiTJoint,
                     depth=48,