From 1e2db9b1e112371ccd3d2498a615a3e09c010f05 Mon Sep 17 00:00:00 2001 From: Henry Addison Date: Mon, 5 Aug 2024 20:58:53 +0100 Subject: [PATCH] disable cncsnpp and cunet from sampling too --- bin/predict.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/bin/predict.py b/bin/predict.py index 18b1b4828..342670a4b 100644 --- a/bin/predict.py +++ b/bin/predict.py @@ -34,17 +34,17 @@ import ml_downscaling_emulator.score_sde_pytorch.models as models # noqa: F401 from ml_downscaling_emulator.score_sde_pytorch.models import utils as mutils -from ml_downscaling_emulator.score_sde_pytorch.models import cncsnpp # noqa: F401 -from ml_downscaling_emulator.score_sde_pytorch.models import cunet # noqa: F401 +# from ml_downscaling_emulator.score_sde_pytorch.models import cncsnpp # noqa: F401 +# from ml_downscaling_emulator.score_sde_pytorch.models import cunet # noqa: F401 from ml_downscaling_emulator.score_sde_pytorch.models import det_cunet # noqa: F401 -from ml_downscaling_emulator.score_sde_pytorch.models import ( # noqa: F401 - layerspp, # noqa: F401 -) # noqa: F401 -from ml_downscaling_emulator.score_sde_pytorch.models import layers # noqa: F401 -from ml_downscaling_emulator.score_sde_pytorch.models import ( # noqa: F401 - normalization, # noqa: F401 -) # noqa: F401 +# from ml_downscaling_emulator.score_sde_pytorch.models import ( # noqa: F401 +# layerspp, # noqa: F401 +# ) # noqa: F401 +# from ml_downscaling_emulator.score_sde_pytorch.models import layers # noqa: F401 +# from ml_downscaling_emulator.score_sde_pytorch.models import ( # noqa: F401 +# normalization, # noqa: F401 +# ) # noqa: F401 import ml_downscaling_emulator.score_sde_pytorch.sampling as sampling from ml_downscaling_emulator.score_sde_pytorch.sde_lib import (