Files
2023-11-14 20:38:44 +06:00
..
2023-11-12 23:37:54 +00:00
2023-11-14 20:38:44 +06:00
2023-11-12 23:24:31 +00:00

Dataset formats

https://github.com/higgsfield-ai/tutorials/raw/main/dataset_upload_tutorial_1080.mp4

We support the following dataset formats:

  • Prompt Completion format
  • The ChatGPT format
  • The Plain Text format

Before uploading your dataset to Hugging Face, please make sure that it is in one of the above formats.

We provide a tutorial on how to convert your dataset to each format.

Upload your datasets to Hugging Face:

from datasets import Dataset
dataset = Dataset.from_dict(format)
dataset.push_to_hub("<HUGGING_FACE_DATASET_REPO>", token="<HUGGING_FACE_TOKEN>") # Example: 'test/test_dataset'

Format: Prompt Completion

prompt_completion = {
        "prompt": [
            "prompt1",
            "prompt2",
        ],
        "completion": [
            "completion1",
            "completion2",
        ]
    }

Format: ChatGPT

chatgpt_format = {
    "chatgpt": [
        [
            {"role": "system", "content": "You are a human."},
            {"role": "user", "content": "No I am not."},
            {"role": "assistant", "content": "I am a robot."},
        ],
    ]
}

Format: Text

text_format = {
    "text": ["text"]
}