From 42091f1800eb41bbeb4bc6e0a257f21956064a98 Mon Sep 17 00:00:00 2001 From: cloudchrischan Date: Mon, 1 Dec 2025 11:23:44 -0500 Subject: [PATCH] Update hstu.py Fix broken link --- recml/layers/keras/hstu.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recml/layers/keras/hstu.py b/recml/layers/keras/hstu.py index 91b889c..de5ecbe 100644 --- a/recml/layers/keras/hstu.py +++ b/recml/layers/keras/hstu.py @@ -35,7 +35,7 @@ def default_bucketization(tensor: Tensor, bucket_size: float = 0.301): """Default bucketization function for RelativeBucketedTimeAndPositionBasedBias. From the original paper's code base: - https://github.com/facebookresearch/generative-recommenders/blob/main/modeling/sequential/hstu.py#L79 + https://github.com/meta-recsys/generative-recommenders/blob/main/generative_recommenders/research/modeling/sequential/hstu.py#L87 Args: tensor: The tensor of timestamps to bucketize.