okayyyyyyy

1 files changed, 1 insertions, 1 deletions

diff --git a/model.py b/model.py
index ff35baa..8592ddc 100644
--- a/model.py
+++ b/model.py
@@ -378,7 +378,7 @@ class PrimedGenerator(Runner):
                 ratio = i / (n-1)
                 a = sub_sequence_a[:, j].float() * (1-ratio)
                 b = sub_sequence_b[:, j].float() * ratio
-                tmp_sub_sequence[:, j] = clamp(a + b, 0, q_levels).long()
+                tmp_sub_sequence[:, j] = torch.clamp(a + b, 0, q_levels).long()
 
             return tmp_sub_sequence