InvokeAI/environment-mac.yml
2022-10-16 20:36:33 +02:00

60 lines
1.4 KiB
YAML

name: invokeai
channels:
- pytorch
- conda-forge
dependencies:
- python>=3.10, <3.11
# 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/invokeai/invokeai-updated/;20,99s/- ([^=]+)==.+/- \1/' environment-mac.yml > environment-mac-updated.yml
# CONDA_SUBDIR=osx-arm64 conda env create -f environment-mac-updated.yml && conda list -n invokeai-updated | awk ' {print " - " $1 "==" $2;} '
# ```
- albumentations
- coloredlogs
- einops
- grpcio
- humanfriendly
- imageio
- imageio-ffmpeg
- imgaug
- kornia
- mpmath
- nomkl
- numpy
- omegaconf
- openh264
- onnx
- onnxruntime
- pudb
- pytorch-lightning
- scipy
- streamlit
- sympy
- tensorboard
- torchmetrics
- pip:
- flask
- flask_socketio
- flask_cors
- dependency_injector
- eventlet
- protobuf
- realesrgan
- send2trash
- test-tube
- transformers
- torch-fidelity
- -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 git+https://github.com/TencentARC/GFPGAN.git#egg=gfpgan
- -e .
variables:
PYTORCH_ENABLE_MPS_FALLBACK: 1