wassname/peft
1
0
Fork 0
mirror of https://github.com/wassname/peft.git synced 2026-09-09 11:28:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update peft_lora_clm_accelerate_ds_zero3_offload.py

Browse Source
This commit is contained in:
Sourab Mangrulkar
2023-02-08 13:35:10 +05:30
parent 537c971a47
commit 83ded43ee7
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/causal_language_modeling/peft_lora_clm_accelerate_ds_zero3_offload.py
+3 -1
View File
@@ -343,8 +343,10 @@ def main():
accelerator.wait_for_everyone()
model.push_to_hub(
f"smangrul/{dataset_name}_{model_name_or_path}_{peft_config.peft_type}_{peft_config.task_type}",
"smangrul/"
+ f"{dataset_name}_{model_name_or_path}_{peft_config.peft_type}_{peft_config.task_type}".replace("/", "_"),
state_dict=accelerator.get_state_dict(model),
use_auth_token=True,
)
accelerator.wait_for_everyone()