File size: 413 Bytes
49538bc
 
 
 
 
774d778
 
49538bc
 
 
ac23842
1
2
3
4
5
6
7
8
9
10
11
numpy==2.0.1
torch==2.4.0
einops==0.7.0 # used in implementation of dual attention / relational attention
huggingface-hub==0.23.4 # for loading pre-trained checkpoints
bertviz==1.4.0 # for visualization
ipykernel==6.29.5 # for bertviz
ipython==8.26.0
tiktoken==0.7.0 # tokenizer used for LM experiments
gradio==4.40.0 # for inference/visualization apps
tqdm==4.66.5
dual_attention==0.0.4  # dual attention package