update requirements to actually work

readme.md

@@ -7,6 +7,10 @@ The purpose of this project is to build a simple large language model from scrat
 This repo is following the guide from https://www.youtube.com/watch?v=UU1WVnMk4E8
 
 In this repo:
-- ./requirements.txt - do `pip install -r requirements.txt`
 - ./shakespeare.txt - This is a sample text used for training a smaller scale model
 - ./bigram_testing.sync.ipynb - This notebook is where I test a basic BiGram model
+
+
+### Prepare environment
+`pip install -r ./requirements-base.txt`
+`pip install -r ./requirements-pytorch.txt`

requirements-base.txt

@@ -0,0 +1,5 @@
+matplotlib
+numpy
+pylzma
+ipykernel
+jupyter

requirements.txt → requirements-pytorch.txt

@@ -1,8 +1,3 @@
-matplotlib
-numpy
-pylzma
-ipykernel
-jupyter
 
 --index-url https://download.pytorch.org/whl/cu118
 torch