Skip to content

Commit

Permalink
CI: remove cuda dependent packages to reduce build time
Browse files Browse the repository at this point in the history
  • Loading branch information
wkpark committed Feb 2, 2024
1 parent 71336d0 commit 65e553f
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions environment-bnb.yml
Original file line number Diff line number Diff line change
Expand Up @@ -7,10 +7,10 @@ channels:

dependencies:
- python
- accelerate
- einops
#- accelerate
#- einops
- scipy
- transformers
#- transformers
- pytest
- pytest-cases
- ipython
Expand Down

0 comments on commit 65e553f

Please sign in to comment.