From 28fa5d64f72006dc475474fba665138751c2045e Mon Sep 17 00:00:00 2001 From: VALADI K JAGANATHAN Date: Fri, 15 Dec 2023 17:14:17 +0530 Subject: [PATCH] Update inference.py --- libs/dance_diffusion/dd/inference.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/libs/dance_diffusion/dd/inference.py b/libs/dance_diffusion/dd/inference.py index f639338..dcb4794 100644 --- a/libs/dance_diffusion/dd/inference.py +++ b/libs/dance_diffusion/dd/inference.py @@ -284,10 +284,10 @@ class DDInference(InferenceBase): else: return output[:, :, -half_chunk_size:] from typing import Tuple, Callable -from diffusion_library.scheduler import SchedulerType -from diffusion_library.sampler import SamplerType -from dance_diffusion.base.model import ModelWrapperBase -from dance_diffusion.base.inference import InferenceBase +from libs.diffusion_library.scheduler import SchedulerType +from libs.diffusion_library.sampler import SamplerType +from libs.dance_diffusion.base.model import ModelWrapperBase +from libs.dance_diffusion.base.inference import InferenceBase from libs.util.util import tensor_slerp_2D, PosteriorSampling