chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-05 02:20:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix minigpt-env wrong torch deps

Browse Source
This commit is contained in:
sharpbai 2023-04-22 02:12:10 +08:00
parent d93f715165
commit 0a781bf14a
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
environment.yml
View File

@ -3,14 +3,15 @@ channels:
- pytorch
- defaults
- anaconda
- nvidia
dependencies:
- python=3.9
- cudatoolkit
- pip
- pytorch=1.12.1
- pytorch=1.13.1
- pytorch-mutex=1.0=cuda
- torchaudio=0.12.1
- torchvision=0.13.1
- torchaudio=0.13.1
- torchvision=0.14.1
- pip:
- accelerate==0.16.0
- aiohttp==3.8.4