瀏覽代碼

Fix typo in generation_algorithms.py (#364)

Ikko Eltociear Ashimine 2 年之前
父節點
當前提交
fd30f7ce10
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      src/petals/utils/generation_algorithms.py

+ 1 - 1
src/petals/utils/generation_algorithms.py

@@ -16,7 +16,7 @@ class DecodingAlgorithm(ABC):
     @abstractmethod
     def __call__(self, logits: torch.Tensor) -> Tuple[TokenIds, HypoIds]:
         """
-        :param logits: A tensor of shape (batch_size, seq_lenth, vocab_size)
+        :param logits: A tensor of shape (batch_size, seq_length, vocab_size)
         :return: A tuple of selected token ids and corresponding hypotheses.
         The shape of the token ids is (batch_size, seq_length), and the shape of the hypotheses is (batch_size)
         """