BK-Lee commited on
Commit
fd49342
1 Parent(s): 2dc271a
Files changed (2) hide show
  1. README.md +1 -1
  2. requirements.txt +2 -1
README.md CHANGED
@@ -6,7 +6,7 @@ colorTo: pink
6
  sdk: gradio
7
  sdk_version: 4.31.5
8
  app_file: app.py
9
- pinned: false
10
  python_version: 3.10.13
11
  pytorch_version: 2.2.0
12
  ---
 
6
  sdk: gradio
7
  sdk_version: 4.31.5
8
  app_file: app.py
9
+ pinned: true
10
  python_version: 3.10.13
11
  pytorch_version: 2.2.0
12
  ---
requirements.txt CHANGED
@@ -9,7 +9,8 @@ wandb
9
  einops
10
  einops_exts
11
  sentencepiece
12
- causal-conv1d
 
13
  mamba-ssm
14
  timm
15
  shortuuid
 
9
  einops
10
  einops_exts
11
  sentencepiece
12
+ git+https://github.com/Dao-AILab/causal-conv1d
13
+ git+https://github.com/Dao-AILab/flash-attention
14
  mamba-ssm
15
  timm
16
  shortuuid