Skip to content

Commit 2aa8393

Browse files
authored
set PYTORCH_ENABLE_MPS_FALLBACK in mac environment (#232)
- this enables cpu fallback for op not yet implemented for m1 gpu
1 parent 6444402 commit 2aa8393

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

environment-mac.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,3 +30,5 @@ dependencies:
3030
- -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
3131
- -e git+https://github.com/lstein/k-diffusion.git@master#egg=k-diffusion
3232
- -e .
33+
variables:
34+
PYTORCH_ENABLE_MPS_FALLBACK: 1

0 commit comments

Comments
 (0)