aducsdr commited on
Commit
aef4a65
·
verified ·
1 Parent(s): e8726ee

Update environment.yml

Browse files
Files changed (1) hide show
  1. environment.yml +23 -23
environment.yml CHANGED
@@ -11,11 +11,11 @@ channels:
11
  # Lista de todas as dependências
12
  dependencies:
13
  # Pacotes principais instalados via Conda para máxima compatibilidade com a GPU
14
- - python=3.10
15
- - pytorch=2.1.0
16
- - torchvision=0.16.0
17
- - torchaudio=2.1.0
18
- - pytorch-cuda=12.1 # Garante que o PyTorch seja compatível com a CUDA da imagem Docker
19
  - pip # Instala o gerenciador de pacotes pip dentro do ambiente
20
 
21
  # Pacotes instalados via Pip
@@ -26,28 +26,28 @@ dependencies:
26
  - python-multipart
27
 
28
  # Dependências do projeto SeedVR
29
- - accelerate==0.24.1
30
- - av==11.0.0
31
- - beautifulsoup4==4.12.2
32
- - controlnet_aux==0.0.7
33
- - diffusers==0.26.3
34
- - einops==0.7.0
35
- - ftfy==6.1.1
36
  - huggingface_hub
37
- - iopath==0.1.10
38
  - mediapy
39
- - omegaconf==2.3.0
40
  - opencv-python
41
- - pandas==2.1.3
42
- - pytorch-lightning==2.1.0
43
- - pytorchvideo==0.1.5
44
- - regex==2023.10.3
45
  - scikit-image
46
- - timm==0.9.10
47
- - tqdm==4.66.1
48
- - transformers==4.35.2
49
- - webdataset==0.2.10
50
- - xformers==0.0.22.post7
51
 
52
  # NOTA IMPORTANTE:
53
  # 'flash_attn' não está listado aqui de propósito.
 
11
  # Lista de todas as dependências
12
  dependencies:
13
  # Pacotes principais instalados via Conda para máxima compatibilidade com a GPU
14
+ - python
15
+ - pytorch
16
+ - torchvision
17
+ - torchaudio
18
+ - pytorch-cuda # Garante que o PyTorch seja compatível com a CUDA da imagem Docker
19
  - pip # Instala o gerenciador de pacotes pip dentro do ambiente
20
 
21
  # Pacotes instalados via Pip
 
26
  - python-multipart
27
 
28
  # Dependências do projeto SeedVR
29
+ - accelerate
30
+ - av
31
+ - beautifulsoup4
32
+ - controlnet_aux
33
+ - diffusers
34
+ - einops
35
+ - ftfy
36
  - huggingface_hub
37
+ - iopath
38
  - mediapy
39
+ - omegaconf
40
  - opencv-python
41
+ - pandas
42
+ - pytorch-lightning
43
+ - pytorchvideo
44
+ - regex
45
  - scikit-image
46
+ - timm
47
+ - tqdm
48
+ - transformers
49
+ - webdataset
50
+ - xformers
51
 
52
  # NOTA IMPORTANTE:
53
  # 'flash_attn' não está listado aqui de propósito.