mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
60be735e80
* Switch to regular pytorch channel and restore Python 3.10 for Macs. Although pytorch-nightly should in theory be faster, it is currently causing increased memory usage and slower iterations: https://github.com/lstein/stable-diffusion/pull/283#issuecomment-1234784885 This changes the environment-mac.yaml file back to the regular pytorch channel and moves the `transformers` dep into pip for now (since it cannot be satisfied until tokenizers>=0.11 is built for Python 3.10). * Specify versions for Pip packages as well.
53 lines
1.4 KiB
YAML
53 lines
1.4 KiB
YAML
name: ldm
|
|
channels:
|
|
- pytorch
|
|
- conda-forge
|
|
dependencies:
|
|
- python==3.10.5
|
|
- pip==22.2.2
|
|
|
|
# pytorch left unpinned
|
|
- pytorch
|
|
- torchvision
|
|
|
|
# I suggest to keep the other deps sorted for convenience.
|
|
# To determine what the latest versions should be, run:
|
|
#
|
|
# ```shell
|
|
# sed -E 's/ldm/ldm-updated/;20,99s/- ([^=]+)==.+/- \1/' environment-mac.yaml > environment-mac-updated.yml
|
|
# CONDA_SUBDIR=osx-arm64 conda env create -f environment-mac-updated.yml && conda list -n ldm-updated | awk ' {print " - " $1 "==" $2;} '
|
|
# ```
|
|
- albumentations==1.2.1
|
|
- coloredlogs==15.0.1
|
|
- einops==0.4.1
|
|
- grpcio==1.46.4
|
|
- humanfriendly==10.0
|
|
- imageio==2.21.2
|
|
- imageio-ffmpeg==0.4.7
|
|
- 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
|
|
- torchmetrics==0.9.3
|
|
- pip:
|
|
- test-tube==0.7.5
|
|
- transformers==4.21.2
|
|
- torch-fidelity==0.3.0
|
|
- -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
|