We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d5e1862 commit 407a43dCopy full SHA for 407a43d
1 file changed
tuning/sft_trainer.py
@@ -292,9 +292,7 @@ def train(
292
)
293
294
# try to load model as a vision model
295
- # model = AutoModelForVision2Seq.from_pretrained(
296
- # model_args.model_name_or_path, **model_kwargs
297
- # )
+ # in transformers v5, AutoModelForVision2Seq was renamed to AutoModelForImageTextToText
298
model = AutoModelForImageTextToText.from_pretrained(
299
model_args.model_name_or_path, **model_kwargs
300
0 commit comments