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