Update loader.py
This commit is contained in:
parent
86884ab446
commit
3275552160
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)
|
||||
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