nbroad
/

hfie-minilm-l6

Feature Extraction

text-embeddings-inference

Model card Files Files and versions

nbroad commited on Jul 12, 2023

Commit

2e00bb1

·

1 Parent(s): cd722d4

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -48,7 +48,7 @@ class EndpointHandler():
         with torch.inference_mode():
             if parameters is None:
-                max_length = None
             else:
                 max_length = parameters.pop("max_length", 512)

         with torch.inference_mode():
             if parameters is None:
+                max_length = 512
             else:
                 max_length = parameters.pop("max_length", 512)