name: ldd
channels:
    - pytorch-nightly
    - nvidia
    - xformers/label/dev
    - defaults
    - conda-forge 
dependencies:
    - cuda-nvcc=12.1.105
    - matplotlib=3.6.2
    - numpy=1.24.3
    - pip=22.3.1
    - python=3.10.8
    - pytorch=2.1.0.dev20230429=*cuda12.1*
    - torchvision=0.16.0.dev20230429
    # - xformers=0.0.19
    - 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
          - bitsandbytes==0.38.1
          - hidet==0.2.3
          - lion-pytorch==0.0.7
          - peft==0.2.0
          - python-slugify>=6.1.2
          - safetensors==0.3.1
          - setuptools==65.6.3
          - test-tube>=0.7.5
          - timm==0.8.17.dev0
          - transformers==4.28.1