SunderAli17 commited on
Commit
a4f6bc0
1 Parent(s): 8a7c697

Create environment.yaml

Browse files
Files changed (1) hide show
  1. environment.yaml +37 -0
environment.yaml ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: SAKBIR
2
+ channels:
3
+ - pytorch
4
+ - nvidia
5
+ - conda-forge
6
+ - defaults
7
+ dependencies:
8
+ - numpy
9
+ - pandas
10
+ - pillow
11
+ - pip
12
+ - python=3.9.15
13
+ - pytorch=2.2.2
14
+ - pytorch-lightning=1.6.5
15
+ - pytorch-cuda=12.1
16
+ - setuptools
17
+ - torchaudio=2.2.2
18
+ - torchmetrics
19
+ - torchvision=0.17.2
20
+ - tqdm
21
+ - pip:
22
+ - accelerate==0.25.0
23
+ - diffusers==0.24.0
24
+ - einops
25
+ - open-clip-torch
26
+ - opencv-python==4.8.1.78
27
+ - tokenizers
28
+ - transformers==4.36.2
29
+ - kornia
30
+ - facenet_pytorch
31
+ - lpips
32
+ - dreamsim
33
+ - pyrallis
34
+ - wandb
35
+ - insightface
36
+ - onnxruntime==1.17.0
37
+ - -e git+https://github.com/openai/CLIP.git@main#egg=clip