GPT_Neo.md

November 11, 2022 ยท View on GitHub

GPT_Neo

You can fine-tune a GPT_Neo model from HuggingFace through model=GPT_Neo, model_path=<hf-or-local-path>, dataset=<dataset-name>.

Example usage:

python run_textbox.py \
    --model=GPT_Neo \
    --model_path=EleutherAI/gpt-neo-125M \
    --dataset=samsum