grammarly
/

brain_query_to_pack_classifier

Model card Files Files and versions

ke-lly-d commited on Jul 17

Commit

a5aea8a

·

1 Parent(s): 458aa66

max token 64

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -24,7 +24,7 @@ class EndpointHandler:
             return_tensors="pt",
             padding=True,
             truncation=True,
-            max_length=128
         )
         inputs = {k: v.to(self.device) for k, v in inputs.items()}

             return_tensors="pt",
             padding=True,
             truncation=True,
+            max_length=64
         )
         inputs = {k: v.to(self.device) for k, v in inputs.items()}