Upload train_infrastructure_model.py with huggingface_hub
Browse files
train_infrastructure_model.py
CHANGED
|
@@ -154,7 +154,6 @@ trainer = SFTTrainer(
|
|
| 154 |
train_dataset=train_dataset,
|
| 155 |
eval_dataset=eval_dataset,
|
| 156 |
peft_config=peft_config,
|
| 157 |
-
tokenizer=tokenizer,
|
| 158 |
args=training_args,
|
| 159 |
max_seq_length=2048,
|
| 160 |
dataset_text_field="text",
|
|
|
|
| 154 |
train_dataset=train_dataset,
|
| 155 |
eval_dataset=eval_dataset,
|
| 156 |
peft_config=peft_config,
|
|
|
|
| 157 |
args=training_args,
|
| 158 |
max_seq_length=2048,
|
| 159 |
dataset_text_field="text",
|