MiniGPT-4/environment.yml

36 lines
666 B
YAML
Raw Normal View History

name: minigptv
2023-04-16 22:04:16 +00:00
channels:
- pytorch
- defaults
- anaconda
dependencies:
- python=3.9
- cudatoolkit
- pip
- pip:
2023-10-13 18:24:10 +00:00
- torch==2.0.0
- torchaudio
- torchvision
2023-10-13 00:14:35 +00:00
- huggingface-hub==0.18.0
2023-04-16 22:04:16 +00:00
- matplotlib==3.7.0
- psutil==5.9.4
2023-10-13 18:24:10 +00:00
- iopath
2023-04-16 22:04:16 +00:00
- pyyaml==6.0
- regex==2022.10.31
- tokenizers==0.13.2
- tqdm==4.64.1
2023-10-13 18:24:10 +00:00
- transformers==4.30.0
2023-04-16 22:04:16 +00:00
- timm==0.6.13
- webdataset==0.2.48
- omegaconf==2.3.0
- opencv-python==4.7.0.72
- decord==0.6.0
2023-10-13 00:14:35 +00:00
- peft==0.2.0
2023-04-16 22:04:16 +00:00
- sentence-transformers
2023-10-13 00:14:35 +00:00
- gradio==3.47.1
2023-10-13 18:24:10 +00:00
- accelerate==0.20.3
- bitsandbytes==0.37.0
- scikit-image
- visual-genome
2023-04-16 22:04:16 +00:00
- wandb