disable cncsnpp and cunet from sampling too

henryaddison · Aug 5, 2024 · ca6c88e · ca6c88e
1 parent bd3dfe3
commit ca6c88e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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