diff --git a/environment-mac.yaml b/environment-mac.yaml index 42a5c359e3..be63a05540 100644 --- a/environment-mac.yaml +++ b/environment-mac.yaml @@ -30,3 +30,5 @@ dependencies: - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers - -e git+https://github.com/lstein/k-diffusion.git@master#egg=k-diffusion - -e . +variables: + PYTORCH_ENABLE_MPS_FALLBACK: 1 diff --git a/ldm/simplet2i.py b/ldm/simplet2i.py index 645f84af53..b83280471d 100644 --- a/ldm/simplet2i.py +++ b/ldm/simplet2i.py @@ -124,7 +124,7 @@ class T2I: grid=False, width=512, height=512, - sampler_name='klms', + sampler_name='k_lms', latent_channels=4, downsampling_factor=8, ddim_eta=0.0, # deterministic