might2901 commited on
Commit
06fa73c
·
verified ·
1 Parent(s): c6a64b8

Add datasets and tqdm to chute image deps

Browse files
Files changed (1) hide show
  1. chute_config.yml +1 -1
chute_config.yml CHANGED
@@ -2,7 +2,7 @@ Image:
2
  from_base: parachutes/python:3.12
3
  set_workdir: /app
4
  run_command:
5
- - pip install torch torchaudio transformers==4.51.3 accelerate==1.6.0 diffusers peft librosa numba scipy pyyaml soundfile ml-collections
6
 
7
  NodeSelector:
8
  gpu_count: 1
 
2
  from_base: parachutes/python:3.12
3
  set_workdir: /app
4
  run_command:
5
+ - pip install torch torchaudio transformers==4.51.3 accelerate==1.6.0 diffusers peft librosa numba scipy pyyaml soundfile ml-collections datasets==3.5.0 tqdm
6
 
7
  NodeSelector:
8
  gpu_count: 1