From bb56c7aef7deff3faa0e31b1a34e615db1a85768 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Fri, 5 Apr 2024 00:21:59 -0500 Subject: [PATCH] Remove unnecessary import as --- animatediff/sampling.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/animatediff/sampling.py b/animatediff/sampling.py index 15d4ce9..82d711e 100644 --- a/animatediff/sampling.py +++ b/animatediff/sampling.py @@ -8,7 +8,6 @@ from einops import rearrange import comfy.ldm.modules.attention as attention from comfy.ldm.modules.diffusionmodules import openaimodel -import comfy.model_management as model_management import comfy.model_management import comfy.samplers import comfy.sample @@ -344,7 +343,7 @@ def motion_sample_factory(orig_comfy_sample: Callable, is_custom: bool=False) -> iter_kwargs = {} if iter_opts.need_sampler: # -5 for sampler_name (not custom) and sampler (custom) - model_management.load_model_gpu(model) + comfy.model_management.load_model_gpu(model) if is_custom: iter_kwargs[IterationOptions.SAMPLER] = None #args[-5] else: