delete extra in of

ChiSym · Jun 17, 2024 · 09f4f69 · 09f4f69
1 parent 0e3be0f
commit 09f4f69
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/genparse/lm.py b/genparse/lm.py
@@ -268,7 +268,7 @@ async def p_next(self, xs, top=None):
 
         _logp = await self._model.next_token_logprobs(tokens)
         _logp = _logp.cpu().numpy() if hasattr(_logp, 'cpu') else _logp
-        _p = np.exp(_logp.cpu().numpy())
+        _p = np.exp(_logp)
 
         assert top is None
         return LazyProb(_p, self._encode, self._decode)