Fix: repeated step: emb hour

ECMWFCode4Earth · Oct 3, 2023 · 207c7e0 · 207c7e0
1 parent 7f72e44
commit 207c7e0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/deepr/validation/generate_data.py b/deepr/validation/generate_data.py
@@ -142,10 +142,9 @@ def predict_xr(
             "sample) at low-res & high-res orography.",
         }
     elif isinstance(model, cDDPMPipeline):
-        hour_emb = get_hour_embedding(times[:, :1], "class", 24).to(model.device)
         prediction = model(
             images=era5,
-            class_labels=hour_emb,
+            class_labels=times[:, :1].to(model.device),
             eta=config["eta"],
             num_inference_steps=config["inference_steps"],
             generator=torch.manual_seed(config.get("seed", 2023)),