generated from fastai/nbdev_template
-
Notifications
You must be signed in to change notification settings - Fork 2.2k
Closed
Description
I run examples you provided here, it says AttributeError: 'NoneType' object has no attribute 'model_init_kwargs'.
Then I also run the example you provided in the README, like this:
# imports
from datasets import load_dataset
from trl import SFTTrainer
# get dataset
dataset = load_dataset("imdb", split="train")
# get trainer
trainer = SFTTrainer(
"facebook/opt-350m",
train_dataset=dataset,
dataset_text_field="text",
max_seq_length=512,
)
# train
trainer.train()
still the same error. Any plan to update these examples?
Metadata
Metadata
Assignees
Labels
No labels