0Tick/e621TagAutocomplete

1---2language:3- en4license: mit5library_name: transformers6tags:7- generated_from_trainer8datasets:9- 0Tick/E621-Random-PostsTag-Scrape10metrics:11- accuracy12co2_eq_emissions: 10013pipeline_tag: text-generation14base_model: distilgpt215model-index:16- name: e621TagAutocomplete17  results: []18---19 20## Model description21 22This is a fine-tuned version of [distilgpt2](https://huggingface.co/distilgpt2) which is intended to be used with the [promptgen](https://github.com/AUTOMATIC1111/stable-diffusion-webui-promptgen) extension inside the AUTOMATIC1111 WebUI.23It is trained on the raw tags of e621 with underscores and spaces24 25 26# Training27 28This model is a fine-tuned version of [distilgpt2](https://huggingface.co/distilgpt2) on a dataset of the tags of 116k random posts of e621.net.29It achieves the following results on the evaluation set:30- Loss: 4.398331- Accuracy: 0.386532 33 34## Training and evaluation data35 36 37Use this collab notebook to train your own model. Also used to train this model38[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/0Tick/stable-diffusion-tools/blob/main/distilgpt2train.ipynb)39 40### Training hyperparameters41 42The following hyperparameters were used during training:43- learning_rate: 5e-0544- train_batch_size: 645- eval_batch_size: 646- seed: 4247- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-0848- lr_scheduler_type: linear49- num_epochs: 3.050 51## Intended uses & limitations52 53Since DistilGPT2 is a distilled version of GPT-2, it is intended to be used for similar use cases with the increased functionality of being smaller and easier to run than the base model. 54 55The developers of GPT-2 state in their [model card](https://github.com/openai/gpt-2/blob/master/model_card.md) that they envisioned GPT-2 would be used by researchers to better understand large-scale generative language models, with possible secondary use cases including: 56 57> - *Writing assistance: Grammar assistance, autocompletion (for normal prose or code)*58> - *Creative writing and art: exploring the generation of creative, fictional texts; aiding creation of poetry and other literary art.*59> - *Entertainment: Creation of games, chat bots, and amusing generations.*60 61Using DistilGPT2, the Hugging Face team built the [Write With Transformers](https://transformer.huggingface.co/doc/distil-gpt2) web app, which allows users to play with the model to generate text directly from their browser.62 63#### Out-of-scope Uses64 65OpenAI states in the GPT-2 [model card](https://github.com/openai/gpt-2/blob/master/model_card.md): 66 67> Because large-scale language models like GPT-2 do not distinguish fact from fiction, we don’t support use-cases that require the generated text to be true.68>69> Additionally, language models like GPT-2 reflect the biases inherent to the systems they were trained on, so we do not recommend that they be deployed into systems that interact with humans unless the deployers first carry out a study of biases relevant to the intended use-case.70 71 72 73### Framework versions74 75- Transformers 4.27.0.dev076- Pytorch 1.13.1+cu11677- Datasets 2.9.078- Tokenizers 0.13.2