-
Notifications
You must be signed in to change notification settings - Fork 488
[SDK] fix issue with serialization of LoraConfig #2567
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Sorry for the late review @Hexoplon!
overall lgtm.
/assign @helenxie-bit @Electronic-Waste
Thank you @andreyvelich, I have updated the code with your suggestions |
@Hexoplon Can you fix CI please ? |
Signed-off-by: Zackeus Bengtsson <github@zackeus.no>
Fixed now, all pre commit hooks are passing |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
/lgtm
/approve
[APPROVALNOTIFIER] This PR is APPROVED This pull-request has been approved by: andreyvelich The full list of commands accepted by this bot can be found here. The pull request process is described here
Needs approval from an approver in each of these files:
Approvers can indicate their approval by writing |
What this PR does / why we need it:
Improve JSON serialization to fix issue occuring when following https://www.kubeflow.org/docs/components/katib/user-guides/llm-hp-optimization/, among other guides. Currently, running the example code with the latest versions of katib sdk peft and transformers, would result in an error of
TypeError: Object of type LoraRuntimeConfig is not JSON serializable
. Now, it will correctly use .to_dict when available, and dict otherwise for non TrainingArgument types