Enavle MPS fallback

README.md

@@ -25,6 +25,7 @@ All credit & copyright goes to https://github.com/lllyasviel.
 * Added support for `onnxruntime` to speed-up DWPose (see the Q&A)
 * Fixed TypeError: expected size to be one of int or Tuple[int] or Tuple[int, int] or Tuple[int, int, int], but got size with types [<class 'numpy.int64'>, <class 'numpy.int64'>] (https://github.com/Fannovel16/comfyui_controlnet_aux/issues/2) (PR: https://github.com/Fannovel16/comfyui_controlnet_aux/pull/71)
 * Fixed ImageGenResolutionFromImage mishape (https://github.com/Fannovel16/comfyui_controlnet_aux/pull/74)
+* Fixed LeRes and MiDaS's incomatipility with MPS device
 
 # Q&A:
 * Why some nodes doesn't appear after I installed this repo?

__init__.py

@@ -15,6 +15,12 @@
     sys.path.insert(0, str(Path(here, "src", pkg_name).resolve()))
 print(f"Registered sys.path: {sys.path}")
 
+#Enable CPU fallback for ops not being supported by MPS like upsample_bicubic2d.out
+#https://github.com/pytorch/pytorch/issues/77764
+#https://github.com/Fannovel16/comfyui_controlnet_aux/issues/2#issuecomment-1763579485
+os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = 1 
+
+
 def load_nodes():
     shorted_errors = []
     full_error_messages = []