Skip to content

Allow JSONL for fine-tuning datasets#486

Merged
squeakymouse merged 2 commits intomainfrom
katiewu/fine-tuning-jsonl
Apr 9, 2024
Merged

Allow JSONL for fine-tuning datasets#486
squeakymouse merged 2 commits intomainfrom
katiewu/fine-tuning-jsonl

Conversation

@squeakymouse
Copy link
Copy Markdown
Contributor

Pull Request Summary

Change gateway validation for fine-tuning datasets: allow JSONL format, and the column names input and output

Test Plan and Usage Guide

Test deployment

@squeakymouse squeakymouse requested a review from a team April 9, 2024 00:44
@@ -52,6 +53,7 @@ def ensure_model_name_is_valid_k8s_label(model_name: str):
def read_csv_headers(file_location: str):
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

update the function name that it now loads other file formats?

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hm I think this function name is still accurate because it treats all files as CSVs, even when they're not? 😛

@squeakymouse squeakymouse enabled auto-merge (squash) April 9, 2024 04:06
@squeakymouse squeakymouse merged commit 8523141 into main Apr 9, 2024
@squeakymouse squeakymouse deleted the katiewu/fine-tuning-jsonl branch April 9, 2024 17:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants