mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
bf50ab9dd6
- move all device init logic to T2I.__init__ - handle m1 specific edge case with autocast device type - check torch.cuda.is_available before using cuda
35 lines
838 B
YAML
35 lines
838 B
YAML
name: ldm
|
|
channels:
|
|
- apple
|
|
- conda-forge
|
|
- pytorch-nightly
|
|
- defaults
|
|
dependencies:
|
|
- python=3.10.4
|
|
- pip=22.1.2
|
|
- pytorch
|
|
- torchvision
|
|
- numpy=1.23.1
|
|
- pip:
|
|
- albumentations==0.4.6
|
|
- opencv-python==4.6.0.66
|
|
- pudb==2019.2
|
|
- imageio==2.9.0
|
|
- imageio-ffmpeg==0.4.2
|
|
- pytorch-lightning==1.4.2
|
|
- omegaconf==2.1.1
|
|
- test-tube>=0.7.5
|
|
- streamlit==1.12.0
|
|
- pillow==9.2.0
|
|
- einops==0.3.0
|
|
- torch-fidelity==0.3.0
|
|
- transformers==4.19.2
|
|
- torchmetrics==0.6.0
|
|
- kornia==0.6.0
|
|
- -e git+https://github.com/openai/CLIP.git@main#egg=clip
|
|
- -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
|
|
- -e git+https://github.com/Birch-san/k-diffusion.git@mps#egg=k_diffusion
|
|
- -e .
|
|
variables:
|
|
PYTORCH_ENABLE_MPS_FALLBACK: 1
|