InvokeAI/scripts/invoke.py
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

5 lines
67 B
Python
Executable File

#!/usr/bin/env python
import ldm.invoke.CLI
ldm.invoke.CLI.main()