disable cncsnpp and cunet from sampling too

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