diff --git a/README.md b/README.md index d009900..669cbe8 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,19 @@ # A Hackers' Guide to Language Models The notebook from the video is [lm-hackers.ipynb](lm-hackers.ipynb). The other files are for axolotl fine-tuning. + +# Setup + +Install the requirements using: + +``` +pip install -r requirements.txt +``` + +# OpenAI API Key + +Get an API key from https://platform.openai.com and place it in a `.env` file: + +``` +OPENAI_API_KEY= +``` \ No newline at end of file diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..94eb139 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,8 @@ +tiktoken +openai +pydantic +fastcore +transformers +wikipedia-api +torch +ipykernel \ No newline at end of file