iw3: Move PYTORCH_ENABLE_MPS_FALLBACK=1 to __init__.py

nagadomi · Aug 13, 2024 · 8c580e1 · 8c580e1
1 parent 463e5f1
commit 8c580e1
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 6 deletions.
diff --git a/.flake8 b/.flake8
@@ -1,8 +1,7 @@
 [flake8]
 ignore = E501 W504 E731
 per-file-ignores =
-     iw3/cli.py: E402
-     iw3/gui.py: E402
+     iw3/__init__.py: E402
      waifu2x/__init__.py: E402
 
 max_line_length = 128

diff --git a/iw3/__init__.py b/iw3/__init__.py
@@ -0,0 +1,2 @@
+import os
+os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1'
diff --git a/iw3/cli.py b/iw3/cli.py
@@ -1,5 +1,3 @@
-import os
-os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1'
 import torch
 from .utils import create_parser, set_state_args, iw3_main
 from . import models # noqa

diff --git a/iw3/gui.py b/iw3/gui.py
@@ -1,5 +1,3 @@
-import os
-os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1'
 import nunif.pythonw_fix  # noqa
 import locale
 import sys
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		import os
		os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1'