Add script for for finetuning a code llm #69
+454
−1
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
iuliaturc
reviewed
Oct 9, 2024
| @@ -83,13 +84,18 @@ def main(): | |||
| retrieved = retriever.invoke(item[args.question_field]) | |||
| item["retrieved"] = [] | |||
| for doc_idx, doc in enumerate(retrieved): | |||
| pprint.pprint(doc) | |||
There was a problem hiding this comment.
These documents can get extremely long. Could you just use logging.info() here, and then set the appropriate logging level whenever you want to see the documents?
There was a problem hiding this comment.
stray print. removing
| @@ -0,0 +1,447 @@ | |||
| import argparse | |||
There was a problem hiding this comment.
Could you leave a comment at the top of this file saying where it was originally forked from and how it's different from the fork?
There was a problem hiding this comment.
I had it below the imports but i can move up
|
Also, it seems like my fix for the github action didn't work. Could you run the following before I get round to fixing it: |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
What does this PR do?
This PR provides basic support for finetuning a code LLM. It is adapted from the following source: https://huggingface.co/learn/cookbook/en/fine_tuning_code_llm_on_single_gpu
NOTE: we still need to add appropriate eval metrics (of which there aren't very many standard ones for finetuning code llms)