diff --git a/modeling/stoppers.py b/modeling/stoppers.py index 8fe5c229..d9f212dd 100644 --- a/modeling/stoppers.py +++ b/modeling/stoppers.py @@ -116,6 +116,7 @@ class Stoppers: return True return False + @staticmethod def adventure_mode_stopper( model: InferenceModel, input_ids: torch.LongTensor,