[partition] fix checking for model files (#1636)

deepjavalibrary · Mar 18, 2024 · 39bc296 · 39bc296
1 parent ce08dfe
commit 39bc296
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/serving/docker/partition/properties_manager.py b/serving/docker/partition/properties_manager.py
@@ -58,7 +58,7 @@ def set_and_validate_model_dir(self):
         if 'model_dir' in self.properties:
             model_dir = self.properties['model_dir']
             model_files = glob.glob(os.path.join(model_dir, '*.bin'))
-            model_files = glob.glob(os.path.join(model_dir, '.safetensors'))
+            model_files = glob.glob(os.path.join(model_dir, '*.safetensors'))
             if not model_files:
                 raise ValueError(
                     f'No .bin or .safetensors files found in the dir: {model_dir}'