psychedelicious a1b1a48fb3 Fixes PYTORCH_ENABLE_MPS_FALLBACK not set correctly ...

`torch` wasn't seeing the environment variable. I suspect this is because it was imported before the variable was set, so was running with a different environment.

Many `torch` ops are supported on MPS so this wasn't noticed immediately, but some samplers like k_dpm_2 still use unsupported operations and need this fallback.

2023-02-04 17:27:33 +11:00

67 B

Executable File

Raw Blame History

View Raw