Skip to content
This repository was archived by the owner on Oct 25, 2024. It is now read-only.

Commit f2b4758

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent ff73cf4 commit f2b4758

File tree

3 files changed

+3
-3
lines changed
  • intel_extension_for_transformers
  • workflows/chatbot/demo/basic_frontend/fastchat/train

3 files changed

+3
-3
lines changed

intel_extension_for_transformers/neural_chat/examples/finetuning/multi_modal/llava_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -288,7 +288,7 @@ def safe_save_model_for_hf_trainer(trainer: transformers.Trainer,
288288
for key, value in state_dict.items()
289289
}
290290
del state_dict
291-
trainer._save(output_dir, state_dict=cpu_state_dict) # noqa
291+
trainer._save(output_dir, state_dict=cpu_state_dict)
292292

293293

294294
def smart_tokenizer_and_embedding_resize(

intel_extension_for_transformers/transformers/modeling/trl_models/modeling_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -729,7 +729,7 @@ def create_reference_model(
729729
param = model.get_parameter(param_name)
730730
param.requires_grad = False
731731

732-
ref_param = ref_model.get_parameter(param_name) # noqa
732+
ref_param = ref_model.get_parameter(param_name)
733733
ref_param = param # noqa
734734

735735
# for all other parameters just make sure they don't use gradients

workflows/chatbot/demo/basic_frontend/fastchat/train/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def safe_save_model_for_hf_trainer(trainer: transformers.Trainer, output_dir: st
7070
if trainer.args.should_save:
7171
cpu_state_dict = {key: value.cpu() for key, value in state_dict.items()}
7272
del state_dict
73-
trainer._save(output_dir, state_dict=cpu_state_dict) # noqa
73+
trainer._save(output_dir, state_dict=cpu_state_dict)
7474

7575

7676
def preprocess(

0 commit comments

Comments
 (0)