test

huggingface · Nov 6, 2022 · ce5ebf3 · ce5ebf3
1 parent 4408e81
commit ce5ebf3
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/inference_server/models/model.py b/inference_server/models/model.py
@@ -85,7 +85,6 @@ def get_downloaded_model_path(model_name: str):
         repo_id=model_name,
         local_files_only=is_offline_mode(),
         cache_dir=os.getenv("TRANSFORMERS_CACHE", None),
-        allow_patterns=["*.pt", "*.bin"],
         # maybe move to safetensors in the future
         ignore_patterns=["*.safetensors", "*log*", "*evaluation*", "tensorboard"],
     )
@@ -97,7 +96,7 @@ def get_downloaded_model_path(model_name: str):
 
 def check_max_input_length(input_token_lengths: List[int], max_input_length: int) -> None:
     if max_input_length is None:
-        return
+        retur
 
     for i in input_token_lengths:
         if i > max_input_length: