Skip to content

Commit 407a43d

Browse files
committed
remove the old commented lines
Signed-off-by: Yash Mehan <yashmehan@gmail.com>
1 parent d5e1862 commit 407a43d

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

tuning/sft_trainer.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -292,9 +292,7 @@ def train(
292292
)
293293
)
294294
# try to load model as a vision model
295-
# model = AutoModelForVision2Seq.from_pretrained(
296-
# model_args.model_name_or_path, **model_kwargs
297-
# )
295+
# in transformers v5, AutoModelForVision2Seq was renamed to AutoModelForImageTextToText
298296
model = AutoModelForImageTextToText.from_pretrained(
299297
model_args.model_name_or_path, **model_kwargs
300298
)

0 commit comments

Comments
 (0)