From ca6c88e9e5c5dbf2614459142c31bbc331c7995c 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 | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/bin/predict.py b/bin/predict.py index 18b1b4828..2144c24fe 100644 --- a/bin/predict.py +++ b/bin/predict.py @@ -34,8 +34,8 @@ 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