awni00 commited on
Commit
774d778
·
verified ·
1 Parent(s): 49538bc

added ipykernel

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -3,6 +3,8 @@ torch==2.4.0
3
  einops==0.7.0 # used in implementation of dual attention / relational attention
4
  huggingface-hub==0.23.4 # for loading pre-trained checkpoints
5
  bertviz==1.4.0 # for visualization
 
 
6
  tiktoken==0.7.0 # tokenizer used for LM experiments
7
  gradio==4.40.0 # for inference/visualization apps
8
  tqdm==4.66.5
 
3
  einops==0.7.0 # used in implementation of dual attention / relational attention
4
  huggingface-hub==0.23.4 # for loading pre-trained checkpoints
5
  bertviz==1.4.0 # for visualization
6
+ ipykernel==6.29.5 # for bertviz
7
+ ipython==8.26.0
8
  tiktoken==0.7.0 # tokenizer used for LM experiments
9
  gradio==4.40.0 # for inference/visualization apps
10
  tqdm==4.66.5