mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
629ca09fda
* Fix conflicts with main branch changes * Fix logic error in choose_autocast_device() that was causing crashes on CUDA systems.
59 lines
1.6 KiB
YAML
59 lines
1.6 KiB
YAML
name: ldm
|
|
channels:
|
|
- pytorch-nightly
|
|
- conda-forge
|
|
dependencies:
|
|
- python==3.9.13
|
|
- pip==22.2.2
|
|
|
|
# pytorch-nightly, left unpinned
|
|
- pytorch
|
|
- torchmetrics
|
|
- torchvision
|
|
|
|
# I suggest to keep the other deps sorted for convenience.
|
|
# If you wish to upgrade to 3.10, try to run this:
|
|
#
|
|
# ```shell
|
|
# CONDA_CMD=conda
|
|
# sed -E 's/python==3.9.13/python==3.10.5/;s/ldm/ldm-3.10/;21,99s/- ([^=]+)==.+/- \1/' environment-mac.yaml > /tmp/environment-mac-updated.yml
|
|
# CONDA_SUBDIR=osx-arm64 $CONDA_CMD env create -f /tmp/environment-mac-updated.yml && $CONDA_CMD list -n ldm-3.10 | awk ' {print " - " $1 "==" $2;} '
|
|
# ```
|
|
#
|
|
# Unfortunately, as of 2022-08-31, this fails at the pip stage.
|
|
- albumentations==1.2.1
|
|
- coloredlogs==15.0.1
|
|
- einops==0.4.1
|
|
- grpcio==1.46.4
|
|
- humanfriendly
|
|
- imageio-ffmpeg==0.4.7
|
|
- imageio==2.21.2
|
|
- imgaug==0.4.0
|
|
- kornia==0.6.7
|
|
- mpmath==1.2.1
|
|
- nomkl
|
|
- numpy==1.23.2
|
|
- omegaconf==2.1.1
|
|
- onnx==1.12.0
|
|
- onnxruntime==1.12.1
|
|
- opencv==4.6.0
|
|
- pudb==2022.1
|
|
- pytorch-lightning==1.6.5
|
|
- scipy==1.9.1
|
|
- streamlit==1.12.2
|
|
- sympy==1.10.1
|
|
- tensorboard==2.9.0
|
|
- transformers==4.21.2
|
|
- pip:
|
|
- invisible-watermark
|
|
- test-tube
|
|
- tokenizers
|
|
- torch-fidelity
|
|
- -e git+https://github.com/huggingface/diffusers.git@v0.2.4#egg=diffusers
|
|
- -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
|
|
- -e git+https://github.com/openai/CLIP.git@main#egg=clip
|
|
- -e git+https://github.com/Birch-san/k-diffusion.git@mps#egg=k_diffusion
|
|
- -e .
|
|
variables:
|
|
PYTORCH_ENABLE_MPS_FALLBACK: 1
|