Fix model_management import so it doesn't get executed twice.

This commit is contained in:
comfyanonymous
2023-04-15 18:55:17 -04:00
parent 81d1f00df3
commit 73c3e11e83
7 changed files with 14 additions and 14 deletions

View File

@@ -3,7 +3,7 @@ from .k_diffusion import external as k_diffusion_external
from .extra_samplers import uni_pc
import torch
import contextlib
import model_management
from comfy import model_management
from .ldm.models.diffusion.ddim import DDIMSampler
from .ldm.modules.diffusionmodules.util import make_ddim_timesteps