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

fix: fall back to auto-detection when user VLM mapping fails

Browse source 
Instead of erroring out when custom_format_mapping fails conversion,
clear it and let auto-detection try. Handles stale cached mappings.
This commit is contained in:
Roland Tannous 2026-03-10 01:42:25 +00:00
commit 8488c2b1df
1 changed files with 9 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

20
studio/backend/utils/datasets/dataset_utils.py
View file

@ -689,17 +689,15 @@ def format_and_template_dataset(
"errors": [],
}
except Exception as e:
errors.append(f"Failed to apply user VLM mapping: {e}")
return {
"dataset": dataset,
"detected_format": "user_mapped",
"final_format": "vlm_conversion_failed",
"is_vlm": True,
"success": False,
"requires_manual_mapping": True,
"warnings": warnings,
"errors": errors,
}
# User mapping failed — fall back to auto-detection instead
# of giving up (handles stale cached mappings gracefully)
warnings.append(
f"User VLM mapping (image='{user_vlm_image_column}', "
f"text='{user_vlm_text_column}') failed: {e} — "
f"falling back to auto-detection"
)
print(f"⚠️ User VLM mapping failed, falling back to auto-detection...")
custom_format_mapping = None # clear so auto-detection runs below
else:
errors.append(
f"Invalid VLM mapping: need 'image' and 'text' roles. Got: {custom_format_mapping}"