-
Notifications
You must be signed in to change notification settings - Fork 1
/
environment.yml
51 lines (51 loc) · 923 Bytes
/
environment.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
name: halc
channels:
- pytorch
- defaults
- anaconda
- main
- nvidia
- conda-forge
dependencies:
- python=3.9
- cudatoolkit-dev=11.7
- pip
- spacy
- spacy-model-en_core_web_sm=3.5.0
- pip:
- torch==2.0.0
- torchaudio
- torchvision
- huggingface-hub==0.20.2
- matplotlib==3.7.0
- psutil==5.9.4
- iopath
- pyyaml==6.0
- regex==2022.10.31
- tokenizers==0.15.0
- tqdm==4.64.1
- -e ./transformers-4.36.2
- timm==0.6.13
- webdataset==0.2.48
- omegaconf==2.3.0
- opencv-python==4.7.0.72
- decord==0.6.0
- peft==0.2.0
- sentence-transformers
- gradio==3.47.1
- accelerate==0.26.1
- bitsandbytes==0.37.0
- scikit-image
- visual-genome
- wandb
- seaborn
- pandas
- shortuuid
- chardet
- openai
- supervision
- addict
- yapf
- pycocotools
- pycocoevalcap
- icecream