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

fix: replace is_dataset_multimodal with is_dataset_image/is_dataset_audio in training orchestrator

Browse source
This commit is contained in:
Roland Tannous 2026-03-08 19:40:00 +00:00
commit a1778d6655
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
studio/backend/core/training/training.py
View file

@ -143,7 +143,8 @@ class TrainingBackend:
"dataset_slice_start": kwargs.get("dataset_slice_start"),
"dataset_slice_end": kwargs.get("dataset_slice_end"),
"custom_format_mapping": kwargs.get("custom_format_mapping"),
"is_dataset_multimodal": kwargs.get("is_dataset_multimodal", False),
"is_dataset_image": kwargs.get("is_dataset_image", False),
"is_dataset_audio": kwargs.get("is_dataset_audio", False),
"num_epochs": kwargs.get("num_epochs", 3),
"learning_rate": kwargs.get("learning_rate", "2e-4"),
"batch_size": kwargs.get("batch_size", 2),