awni00 commited on
Commit
0a3b673
·
verified ·
1 Parent(s): 0372c23

added numpy and updated torch

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -3
requirements.txt CHANGED
@@ -1,7 +1,9 @@
1
- torch==2.2.1
 
2
  einops==0.7.0 # used in implementation of dual attention / relational attention
3
  huggingface-hub==0.23.4 # for loading pre-trained checkpoints
4
  bertviz==1.4.0 # for visualization
5
- tiktoken==0.5.2 # tokenizer used for LM experiments
6
  gradio==4.40.0 # for inference/visualization apps
7
- dual_attention==0.0.3
 
 
1
+ numpy==2.0.1
2
+ 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
+ dual_attention==0.0.3
9
+ tqdm==4.66.5