rivapereira123 commited on
Commit
b52b1ab
·
verified ·
1 Parent(s): 93b6b74

Update finetune_flan_t5.py

Browse files
Files changed (1) hide show
  1. finetune_flan_t5.py +2 -1
finetune_flan_t5.py CHANGED
@@ -1,6 +1,7 @@
1
  from datasets import load_dataset
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, TrainingArguments
3
- from trl import SFTTrainer, DataCollatorForSeq2Seq
 
4
  import torch
5
 
6
  # Load your dataset (from the converted JSONL file)
 
1
  from datasets import load_dataset
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, TrainingArguments
3
+ from trl import SFTTrainer # ✅ from trl
4
+ from transformers import DataCollatorForSeq2Seq # ✅ from transformers
5
  import torch
6
 
7
  # Load your dataset (from the converted JSONL file)