Skip to content

Commit

Permalink
Update requirements_cpu.txt
Browse files Browse the repository at this point in the history
Removing the mamba related requirements from cpu, we should add these to the gpu requirements_gpu.txt.
  • Loading branch information
gkielian authored Feb 25, 2025
1 parent f174102 commit f01f5c6
Showing 1 changed file with 0 additions and 5 deletions.
5 changes: 0 additions & 5 deletions requirements_cpu.txt
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,3 @@ tiktoken==0.7.0
torchinfo==1.8.0
transformers==4.44.2
wandb==0.18.3
einops==0.8.1
mamba-ssm==2.2.4
setuptools==75.8.0
ninja==1.11.1.3
causal-conv1d==1.5.0.post8

0 comments on commit f01f5c6

Please sign in to comment.