Update app.py

app.py

@@ -16,7 +16,7 @@ from transformers import (
 class StoppingCriteriaSub(StoppingCriteria):
     def __init__(self, stops = [], encounters=1):
         super().__init__()
-        # self.stops = [stop.to("cuda") for stop in stops]
+        self.stops = [stop.to("cuda") for stop in stops]
 
     def __call__(self, input_ids: torch.LongTensor, scores: torch.FloatTensor):
         last_token = input_ids[0][-1]