diff --git a/.gitignore b/.gitignore index 47ed773..bfad8c6 100644 --- a/.gitignore +++ b/.gitignore @@ -16,6 +16,8 @@ ch05/01_main-chapter-code/the-verdict.txt ch06/01_main-chapter-code/loss-plot.pdf ch06/01_main-chapter-code/accuracy-plot.pdf +ch07/01_main-chapter-code/loss-plot.pdf + # Checkpoint files appendix-A/01_main-chapter-code/model.pth appendix-E/01_main-chapter-code/gpt2 @@ -30,6 +32,8 @@ ch06/01_main-chapter-code/gpt2 ch06/02_bonus_additional-experiments/gpt2 ch06/03_bonus_imdb-classification/gpt2 +ch07/01_main-chapter-code/gpt2-medium355M-sft.pth + # Datasets appendix-E/01_main-chapter-code/sms_spam_collection.zip appendix-E/01_main-chapter-code/sms_spam_collection diff --git a/ch07/01_main-chapter-code/ch07.ipynb b/ch07/01_main-chapter-code/ch07.ipynb index 59917ed..65acf60 100644 --- a/ch07/01_main-chapter-code/ch07.ipynb +++ b/ch07/01_main-chapter-code/ch07.ipynb @@ -1283,7 +1283,7 @@ }, "outputs": [], "source": [ - "def extract_response(response_text):\n", + "def extract_response(response):\n", " return response[response.find(\"\\n### Response\")+len(\"\\n### Response:\")+1:]" ] },