31

app/requirements.txt

@@ -3,5 +3,4 @@ numpy
 transformers
 wandb
 tqdm
-streamlit
-tiktoken
+streamlit

app/sample.py

@@ -1,6 +1,8 @@
 """
 Sample from a trained model
 """
+pip3 install tiktoken
+
 import os
 import pickle
 from contextlib import nullcontext
@@ -8,8 +10,6 @@ import torch
 import tiktoken
 from model import GPTConfig, GPT
 
-pip install tiktoken
-
 # -----------------------------------------------------------------------------
 init_from = 'resume' # either 'resume' (from an out_dir) or a gpt2 variant (e.g. 'gpt2-xl')
 out_dir = 'out' # ignored if init_from is not 'resume'