From 9e62a2fb3b6db178dda785e4e24ad7f2a0357918 Mon Sep 17 00:00:00 2001 From: Eric Kerfoot Date: Thu, 15 Aug 2024 12:53:16 +0100 Subject: [PATCH] LDM Script Fix Signed-off-by: Eric Kerfoot --- generation/3d_ldm/inference.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/generation/3d_ldm/inference.py b/generation/3d_ldm/inference.py index 6786dd2ba6..19c638a671 100644 --- a/generation/3d_ldm/inference.py +++ b/generation/3d_ldm/inference.py @@ -20,8 +20,8 @@ import nibabel as nib import numpy as np import torch -from generative.inferers import LatentDiffusionInferer -from generative.networks.schedulers import DDPMScheduler +from monai.inferers import LatentDiffusionInferer +from monai.networks.schedulers import DDPMScheduler from monai.config import print_config from monai.utils import set_determinism