From a58bde615de8f7f3bc3f4641b6ff028f2f6e10ee Mon Sep 17 00:00:00 2001
From: Rajat Sen <rsen91@gmail.com>
Date: Wed, 1 Oct 2025 21:47:42 +0000
Subject: [PATCH] minor doc changes

---
 src/timesfm/torch/transformer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/timesfm/torch/transformer.py b/src/timesfm/torch/transformer.py
index b7fba8b..a2c12da 100644
--- a/src/timesfm/torch/transformer.py
+++ b/src/timesfm/torch/transformer.py
@@ -131,7 +131,7 @@ def _dot_product_attention(
 
 def _torch_dot_product_attention(query, key, value, mask=None):
   """
-  Performs the exact same (unscaled) attention as your original function,
+  Performs the exact same (unscaled) attention as the above function,
   but using the fast and fused F.scaled_dot_product_attention kernel.
   """