Hi,
I am not able to finetune Qwen3-4B model by using parameters;
FastVisionModel.for_training(model) # Enable for training!
trainer = SFTTrainer(
model = model,
tokenizer = tokenizer,
data_collator = UnslothVisionDataCollator(model, tokenizer),
train_dataset = processed_dataset,
args = SFTConfig(
per_device_train_batch_size = 2,
gradient_accumulation_steps = 4,
warmup_steps = 5,
max_steps = 30,
learning_rate = 2e-4,
fp16 = not is_bf16_supported(),
bf16 = is_bf16_supported(),
logging_steps = 1,
optim = “adamw_8bit”,
weight_decay = 0.01,
lr_scheduler_type = “linear”,
seed = 3407,
output_dir = “outputs”,
report_to = “none”,
remove_unused_columns = False,
dataset_text_field = “”,
dataset_kwargs = {“skip_prepare_dataset”: True},
dataset_num_proc = 4,
max_seq_length = 2048,
),
)
Returns
TypeError: Unsloth: UnslothVisionDataCollator is only for image models!
Can anyone help me?