Update loader.py
This commit is contained in:
parent
e88ea0ad7f
commit
94be60007a
1 changed files with 1 additions and 1 deletions
|
|
@ -858,7 +858,7 @@ class FastModel(FastBaseModel):
|
|||
trust_remote_code = trust_remote_code,
|
||||
)
|
||||
# Patch it as well!
|
||||
model = FastBaseModel.post_patch_model(model, use_gradient_checkpointing,trust_remote_code = trust_remote_code)
|
||||
model = FastBaseModel.post_patch_model(model, use_gradient_checkpointing, trust_remote_code = trust_remote_code)
|
||||
pass
|
||||
return model, tokenizer
|
||||
pass
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue