We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 902a68c commit e03f989Copy full SHA for e03f989
server/lorax_server/utils/attention/common.py
@@ -49,5 +49,5 @@ def __init__(
49
self.max_k = max_k
50
51
def clamp(self, max):
52
- self.input_lengths = torch.clamp(self.input_lengths, max=max)
+ self.input_lengths.data.clamp_(max=max)
53
return self
0 commit comments