From 842f26654bbe7dfd2f45df1fd2660d3f902af8cc Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 17 Feb 2023 14:53:25 +0100 Subject: Remove xformers, switch to Pytorch Nightly --- environment.yaml | 18 +++++++----------- 1 file changed, 7 insertions(+), 11 deletions(-) (limited to 'environment.yaml') diff --git a/environment.yaml b/environment.yaml index f5632bf..8010c09 100644 --- a/environment.yaml +++ b/environment.yaml @@ -1,28 +1,24 @@ name: ldd channels: - - pytorch + - pytorch-nightly - nvidia - xformers/label/dev - defaults dependencies: - - cudatoolkit=11.3 - - libcufile=1.4.0.31 + - cudatoolkit=11.7 - matplotlib=3.6.2 - numpy=1.23.4 - pip=22.3.1 - python=3.10.8 - - pytorch=1.13.1=*cuda* - - torchvision=0.14.1 + - pytorch=2.0.0.dev20230216=*cuda* + - torchvision=0.15.0.dev20230216 - pip: - -e . - -e git+https://github.com/huggingface/diffusers#egg=diffusers - - -e git+https://github.com/cloneofsimo/lora#egg=lora-diffusion - - accelerate==0.15.0 + - accelerate==0.16.0 - bitsandbytes==0.37.0 - python-slugify>=6.1.2 - - safetensors==0.2.7 + - safetensors==0.2.8 - setuptools==65.6.3 - test-tube>=0.7.5 - - transformers==4.25.1 - - triton==2.0.0.dev20221202 - - xformers==0.0.17.dev443 + - transformers==4.26.1 -- cgit v1.2.3-54-g00ecf