Update inference.py

This commit is contained in:
VALADI K JAGANATHAN
2023-12-15 17:14:17 +05:30
committed by GitHub
parent d3bb7695b2
commit 28fa5d64f7
+4 -4
View File
@@ -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