Yaron Koresh commited on
Commit
4f6c5c8
·
verified ·
1 Parent(s): f024b1a

Update env.yml

Browse files
Files changed (1) hide show
  1. env.yml +34 -1
env.yml CHANGED
@@ -1,4 +1,37 @@
1
  name: env
2
  channels:
3
  - conda-forge
4
- - defaults
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  name: env
2
  channels:
3
  - conda-forge
4
+ - defaults
5
+ dependencies:
6
+ - python
7
+ - pip:
8
+ - datasets
9
+ - huggingface-hub
10
+ - hf-transfer
11
+ - protobuf
12
+ - click
13
+ - pydantic
14
+ - gradio
15
+ - uvicorn
16
+ - spaces
17
+ - accelerate
18
+ - diffusers@git+https://github.com/huggingface/diffusers@main
19
+ - requests
20
+ - numba
21
+ - lxml
22
+ - sentencepiece
23
+ - invisible_watermark
24
+ - huggingface_hub
25
+ - beautifulsoup4
26
+ - bs4
27
+ - peft
28
+ - transformers
29
+ - flax
30
+ - cython
31
+ - numpy
32
+ - mpi4py
33
+ - dill
34
+ - pox
35
+ - pathos
36
+ - "jax[cuda_local] jaxlib[cuda_local] --extra-index-url https://storage.googleapis.com/jax-releases/jax_cuda_releases.html"
37
+ - "torch torchaudio torchvision triton xformers --extra-index-url https://download.pytorch.org/whl/cu121"