github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[CI] downgrade fbgemm. (#1778)

pull/1776/head
Super Daniel 2022-10-31 18:18:45 +08:00 committed by GitHub
parent f34dab4270
commit 5ea89f6456
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
requirements/requirements-test.txt
View File

@ -1,11 +1,12 @@
diffusers diffusers
fbgemm-gpu==0.2.0
pytest pytest
torchvision torchvision
transformers transformers
timm timm
titans titans
torchaudio torchaudio
torchrec torchrec==0.2.0
contexttimer contexttimer
einops einops
triton==2.0.0.dev20221011 triton==2.0.0.dev20221011