re-enable cncsnpp for training

henryaddison · Aug 5, 2024 · 9777319 · 9777319
1 parent 50be46a
commit 9777319
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ml_downscaling_emulator/score_sde_pytorch/run_lib.py b/src/ml_downscaling_emulator/score_sde_pytorch/run_lib.py
@@ -28,7 +28,7 @@
 from codetiming import Timer
 import logging
 # Keep the import below for registering all model definitions
-from .models import det_cunet # cunet, cncsnpp
+from .models import det_cunet, cunet, cncsnpp
 from . import losses
 from .models.location_params import LocationParams
 from . import sampling