miyuki2026 commited on
Commit
fba77e5
·
1 Parent(s): e72bad5
examples/tutorials/dpo/ultrachat/step_2_train_sft_model_deepspeed.py CHANGED
@@ -47,7 +47,7 @@ import transformers
47
  from transformers import AutoModelForCausalLM, AutoTokenizer, DataCollatorForLanguageModeling
48
  from trl import SFTTrainer, SFTConfig, DataCollatorForCompletionOnlyLM
49
 
50
- transformers.logging.set_verbosity_error()
51
 
52
 
53
  def get_args():
 
47
  from transformers import AutoModelForCausalLM, AutoTokenizer, DataCollatorForLanguageModeling
48
  from trl import SFTTrainer, SFTConfig, DataCollatorForCompletionOnlyLM
49
 
50
+ # transformers.logging.set_verbosity_error()
51
 
52
 
53
  def get_args():