Skip to content

Conversation

@YashasviChaurasia
Copy link
Contributor

@YashasviChaurasia YashasviChaurasia commented Nov 25, 2025

Description of the change

The PR adds tuning-config-recommender functionality to fms-hf-tuning simplifying the fine-tuning experience for the end user.

This contributes fms-recommender.py script which simplifies the tuning experience for the user and helps them avoid minor/major pitfalls which can hinder their tuning experience..
With the help of tuning-config-recommender now we can correct erroneous launch commands or get the recommender to provide suggestions for various flags used while fine-tuning.

The fms-recommender.py script has a --preview flag which would preview the suggested command and not execute it..
Note: by default without the --preview flag the suggested command is executed !

Following is a sample of running the command in preview mode:

python fms-recommender.py --preview tuning.sft_trainer \
--model_name_or_path ibm-granite/granite-4.0-h-350m \
--training_data_path tatsu-lab/alpaca \
--tuning_strategy full

Preview command:

image

Generates a valid data_config file with the datasets, chat-template, data_handlers, etc.

image

Generates a valid accelerate_config

image

Note:

Currently the tuning-config-recommender is undergoing review cycles before its first release and hence the pyproject.yaml is not linked to a release version..
for now the pyproject would install via main branch which might not work the same way as shown above due to pending essential PR's for FMSAdapter at : PR's are undergoing review!

The above commands can be replicated via installing recommender from the main branch

Related issue number

How to verify the PR

Was the PR tested

  • I have added >=1 unit test(s) for every new method I have added.
  • I have ensured all unit tests pass

@github-actions
Copy link

Thanks for making a pull request! 😃
One of the maintainers will review and advise on the next steps.

@YashasviChaurasia
Copy link
Contributor Author

updated pyproject to use main branch until first release.
major changes were merged into main for tuning-config-recommender.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants