fix kwarg to cDDPM

ECMWFCode4Earth · Oct 25, 2023 · 761cc3b · 761cc3b
1 parent d5c9c9a
commit 761cc3b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deepr/workflow.py b/deepr/workflow.py
@@ -421,6 +421,7 @@ def generate_predictions(self):
                 hour_embed_type=self.inference_config.get("hour_embed_type", "class"),
                 hour_embed_dim=self.inference_config.get("hour_embed_dim", 64),
                 instance_norm=self.inference_config.get("instance_norm", False),
+                learn_residuals=self.inference_config.get("learn_residuals", False),
             )
             pipe.to(self.inference_config["device"])
             generate_data.generate_validation_dataset(