georgesung's picture
Update README.md
d5f7662
|
raw
history blame
714 Bytes
---
license: apache-2.0
datasets:
- ehartford/wizard_vicuna_70k_unfiltered
---
Fine-tuned [OpenLLaMA-7B](https://huggingface.co/openlm-research/open_llama_7b) with an uncensored/unfiltered Wizard-Vicuna conversation dataset [ehartford/wizard_vicuna_70k_unfiltered](https://huggingface.co/datasets/ehartford/wizard_vicuna_70k_unfiltered).
Used QLoRA for fine-tuning. Trained for one epoch on a 24GB GPU (NVIDIA A10G) instance, took ~18 hours to train.
Code used to train the model is available [here](https://github.com/georgesung/llm_qlora).
Since this was my first time fine-tuning an LLM, I also wrote an accompanying blog post about how I performed the training :)
https://georgesung.github.io/ai/qlora-ift/