diff --git a/ldm/invoke/CLI.py b/ldm/invoke/CLI.py
index 2d673a9112..ca4153f53d 100644
--- a/ldm/invoke/CLI.py
+++ b/ldm/invoke/CLI.py
@@ -4,6 +4,9 @@ import sys
 import shlex
 import traceback
 
+if sys.platform == "darwin":
+    os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
+
 from ldm.invoke.globals import Globals
 from ldm.generate import Generate
 from ldm.invoke.prompt_parser import PromptParser
@@ -21,9 +24,6 @@ import ldm.invoke
 # global used in multiple functions (fix)
 infile = None
 
-if sys.platform == 'darwin':
-    os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
-
 def main():
     """Initialize command-line parsers and the diffusion model"""
     global infile
diff --git a/scripts/invoke.py b/scripts/invoke.py
index 710cea3830..7431300f15 100755
--- a/scripts/invoke.py
+++ b/scripts/invoke.py
@@ -1,7 +1,4 @@
 #!/usr/bin/env python
 
-import sys
-import os
-
 import ldm.invoke.CLI
 ldm.invoke.CLI.main()