ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-05-23 19:30:37 +00:00
commit 8b932c1d9f
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
studio/backend/core/training/trainer.py
View file

@ -3125,9 +3125,13 @@ class UnslothTrainer:
FastVisionModel.for_training(self.model)
vision_image_size = training_args.get("vision_image_size")
if vision_image_size is None:
data_collator = UnslothVisionDataCollator(self.model, self.tokenizer)
data_collator = UnslothVisionDataCollator(
self.model, self.tokenizer
)
else:
logger.info(f"Vision image resize: {vision_image_size} (max dimension)\n")
logger.info(
f"Vision image resize: {vision_image_size} (max dimension)\n"
)
data_collator = UnslothVisionDataCollator(
self.model,
self.tokenizer,