diff --git a/diffuser/models/temporal.py b/diffuser/models/temporal.py
index 92c00973..786a2173 100644
--- a/diffuser/models/temporal.py
+++ b/diffuser/models/temporal.py
@@ -183,8 +183,8 @@ def __init__(
                 Downsample1d(dim_out)
             ]))
 
-            if not is_last:
-                horizon = horizon // 2
+            # if not is_last:
+            horizon = horizon // 2
 
         mid_dim = dims[-1]
         mid_dim_2 = mid_dim // 2