[Frontend] [Core] Integrate Tensorizer in to S3 loading machinery, allow passing arbitrary arguments during save/load (#19619)
Signed-off-by: Sanger Steel <sangersteel@gmail.com> Co-authored-by: Eta <esyra@coreweave.com>
This commit is contained in:
@@ -686,8 +686,11 @@ class ModelConfig:
|
||||
|
||||
# If tokenizer is same as model, download to same directory
|
||||
if model == tokenizer:
|
||||
s3_model.pull_files(
|
||||
model, ignore_pattern=["*.pt", "*.safetensors", "*.bin"])
|
||||
s3_model.pull_files(model,
|
||||
ignore_pattern=[
|
||||
"*.pt", "*.safetensors", "*.bin",
|
||||
"*.tensors"
|
||||
])
|
||||
self.tokenizer = s3_model.dir
|
||||
return
|
||||
|
||||
@@ -695,7 +698,8 @@ class ModelConfig:
|
||||
if is_s3(tokenizer):
|
||||
s3_tokenizer = S3Model()
|
||||
s3_tokenizer.pull_files(
|
||||
model, ignore_pattern=["*.pt", "*.safetensors", "*.bin"])
|
||||
model,
|
||||
ignore_pattern=["*.pt", "*.safetensors", "*.bin", "*.tensors"])
|
||||
self.tokenizer = s3_tokenizer.dir
|
||||
|
||||
def _init_multimodal_config(self) -> Optional["MultiModalConfig"]:
|
||||
|
||||
Reference in New Issue
Block a user