From 74a5974ba30c170198890e59c92463bf5319fe64 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sat, 29 Apr 2023 16:35:41 +0200 Subject: torch.compile won't work yet, keep code prepared --- environment_nightly.yaml | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 environment_nightly.yaml (limited to 'environment_nightly.yaml') diff --git a/environment_nightly.yaml b/environment_nightly.yaml new file mode 100644 index 0000000..4c5c798 --- /dev/null +++ b/environment_nightly.yaml @@ -0,0 +1,31 @@ +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 -- cgit v1.2.3-54-g00ecf