Update pipeline_echomimicv2.py

smthemex · Jan 3, 2025 · 4a399c3 · 4a399c3
1 parent e0933ab
commit 4a399c3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/echomimic_v2/src/pipelines/pipeline_echomimicv2.py b/echomimic_v2/src/pipelines/pipeline_echomimicv2.py
@@ -22,8 +22,8 @@
 from tqdm import tqdm
 
 from ..models.mutual_self_attention import ReferenceAttentionControl
-from ..pipelines.context import get_context_scheduler
-from ..pipelines.utils import get_tensor_interpolation_method
+from .context import get_context_scheduler
+from .utils import get_tensor_interpolation_method
 
 
 @dataclass
@@ -621,4 +621,4 @@ def __call__(
         if not return_dict:
             return images
 
-        return EchoMimicV2PipelineOutput(videos=images)
+        return EchoMimicV2PipelineOutput(videos=images)