diff --git a/bin/predict.py b/bin/predict.py
index 18b1b482..342670a4 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 (