Spaces:
Paused
Paused
Commit ·
fba77e5
1
Parent(s): e72bad5
update
Browse files
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():
|