summaryrefslogtreecommitdiffstats
path: root/environment.yaml
blob: 4a73688b90a31b9f6ac45ea53289a39e36447f02 (plain) (blame)
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
name: ldd
channels:
    - pytorch
    - nvidia
    - xformers/label/dev
    - defaults
    - conda-forge
dependencies:
    - cuda-nvcc=11.8
    - cuda-cudart-dev=11.8
    - gcc=11.3.0
    - gxx=11.3.0
    - matplotlib=3.6.2
    - numpy=1.24.3
    - pip=22.3.1
    - python=3.10.8
    - pytorch=2.0.1=*cuda11.8*
    - scipy=1.10.1
    - torchvision=0.15.2
    - xformers=0.0.21.dev542+git.a205b24
    - pip:
          - -e .
          - -e git+https://github.com/huggingface/accelerate#egg=accelerate
          - -e git+https://github.com/huggingface/diffusers#egg=diffusers
          - -e git+https://github.com/facebookresearch/dadaptation#egg=dadaptation
          - --pre --extra-index-url https://download.hidet.org/whl hidet
          - bitsandbytes==0.39.1
          - lion-pytorch==0.0.7
          - peft==0.3.0
          - prodigyopt==1.0
          - python-slugify>=6.1.2
          - safetensors==0.3.1
          - setuptools==65.6.3
          - test-tube>=0.7.5
          - timm==0.9.2
          - transformers==4.30.1