diff --git a/eval.py b/eval.py index ba27375f9..c7ea2e4f9 100644 --- a/eval.py +++ b/eval.py @@ -141,7 +141,7 @@ def _model_call(self, inps): ) ) x = seq.index_select(0, input_pos).view(1, -1) - with measure_time(message=None as measure: + with measure_time(message=None) as measure: logits = model_forward(self._model, x, input_pos) self.times.append(measure.time()) return logits