Upload lazy_tensor_loader.py
Browse files- lazy_tensor_loader.py +1 -1
lazy_tensor_loader.py
CHANGED
|
@@ -47,7 +47,7 @@ class ShardedTensorIndex:
|
|
| 47 |
for model_file_name in [
|
| 48 |
"model.safetensors",
|
| 49 |
"pytorch_model.bin",
|
| 50 |
-
"pytorch_model
|
| 51 |
]:
|
| 52 |
candidate_path = os.path.join(base_path, model_file_name)
|
| 53 |
if os.path.exists(candidate_path) or os.path.exists(
|
|
|
|
| 47 |
for model_file_name in [
|
| 48 |
"model.safetensors",
|
| 49 |
"pytorch_model.bin",
|
| 50 |
+
"pytorch_model*.safetensors",
|
| 51 |
]:
|
| 52 |
candidate_path = os.path.join(base_path, model_file_name)
|
| 53 |
if os.path.exists(candidate_path) or os.path.exists(
|