From 83ded43ee740d8ca0279994070caa8cfe0453d4c Mon Sep 17 00:00:00 2001 From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com> Date: Wed, 8 Feb 2023 13:35:10 +0530 Subject: [PATCH] Update peft_lora_clm_accelerate_ds_zero3_offload.py --- .../peft_lora_clm_accelerate_ds_zero3_offload.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/causal_language_modeling/peft_lora_clm_accelerate_ds_zero3_offload.py b/examples/causal_language_modeling/peft_lora_clm_accelerate_ds_zero3_offload.py index 1a1b037..1f10f1b 100644 --- a/examples/causal_language_modeling/peft_lora_clm_accelerate_ds_zero3_offload.py +++ b/examples/causal_language_modeling/peft_lora_clm_accelerate_ds_zero3_offload.py @@ -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()