awni00 commited on
Commit
c70c632
·
verified ·
1 Parent(s): 762521e

added dual_attention to requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -3,4 +3,5 @@ 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
 
 
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