From 7301f2250ffa3a21aaf271beebe121acab28e6b3 Mon Sep 17 00:00:00 2001 From: Mohamed Rashad Date: Fri, 13 Jan 2023 17:19:38 +0000 Subject: [PATCH] Update to numpy 1.24.0 --- generation/strategies.py | 6 +++--- requirements.txt | 3 ++- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/generation/strategies.py b/generation/strategies.py index 92525bd..58aa996 100644 --- a/generation/strategies.py +++ b/generation/strategies.py @@ -14,7 +14,7 @@ def __init__(self, batch_size, invalid_slices=[], temperature=1., top_k=200, eps if end_tokens is None: end_tokens = [] self.end_tokens = end_tokens - self._is_done = np.zeros(self.batch_size, dtype=np.bool) + self._is_done = np.zeros(self.batch_size, dtype=np.bool_) @property def is_done(self) -> bool: @@ -43,7 +43,7 @@ def forward(self, logits, tokens, mems, temperature=None): return tokens, mems def finalize(self, tokens, mems): - self._is_done = np.zeros(self.batch_size, dtype=np.bool) + self._is_done = np.zeros(self.batch_size, dtype=np.bool_) return tokens, mems @@ -77,7 +77,7 @@ def _init_cache(self): self.cached_beam_scores = 0 # [batch_size] self.cached_beam_ngram_bans = [[{} for _ in range(self.num_beams)] for _ in range(self.batch_size)] self.length_generated = 0 - self._is_done = np.zeros(self.batch_size, dtype=np.bool) + self._is_done = np.zeros(self.batch_size, dtype=np.bool_) def _add_end_beams(self, score, beam, batch_idx): score = score / ((5.0 + len(beam)) / 6) ** self.length_penalty # Magic number for OpenNMT diff --git a/requirements.txt b/requirements.txt index 8e30f9b..fae21d2 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,4 +3,5 @@ icetk apex scipy dataclass_wizard -cpm_kernels \ No newline at end of file +cpm_kernels +numpy>=1.24.0 \ No newline at end of file