Merge branch 'main' into main
This commit is contained in:
commit
fe770f23c0
1 changed files with 2 additions and 0 deletions
|
|
@ -475,6 +475,7 @@ class ExportBackend:
|
|||
self.current_model.save_pretrained_merged(
|
||||
save_directory,
|
||||
self.current_tokenizer,
|
||||
save_method = "merged_16bit",
|
||||
)
|
||||
else:
|
||||
self.current_model.save_pretrained(save_directory)
|
||||
|
|
@ -510,6 +511,7 @@ class ExportBackend:
|
|||
self.current_model.save_pretrained_merged(
|
||||
tmp_dir,
|
||||
self.current_tokenizer,
|
||||
save_method = "merged_16bit",
|
||||
)
|
||||
self.current_model.push_to_hub_merged(
|
||||
repo_id,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue