Fix scaling that affects constant series (#294)

abdulfatir · web-flow · commit eec771e3394c · 2025-04-07T08:54:03.000+02:00
diff --git a/src/chronos/chronos_bolt.py b/src/chronos/chronos_bolt.py
@@ -88,7 +88,7 @@ def forward(
             scale = torch.nan_to_num(
                 torch.nanmean((x - loc).square(), dim=-1, keepdim=True).sqrt(), nan=1.0
             )
-            scale = torch.where(scale == 0, torch.abs(loc) + self.eps, scale)
+            scale = torch.where(scale == 0, self.eps, scale)
         else:
             loc, scale = loc_scale
 

Original file line number	Diff line number	Diff line change
`@@ -88,7 +88,7 @@ def forward(`
`88`	`88`	`scale = torch.nan_to_num(`
`89`	`89`	`torch.nanmean((x - loc).square(), dim=-1, keepdim=True).sqrt(), nan=1.0`
`90`	`90`	`)`
`91`		`- scale = torch.where(scale == 0, torch.abs(loc) + self.eps, scale)`
	`91`	`+ scale = torch.where(scale == 0, self.eps, scale)`
`92`	`92`	`else:`
`93`	`93`	`loc, scale = loc_scale`
`94`	`94`