Skip to content

Commit

Permalink
Force PYTORCH_MPS_FALLBACK to be set to 1 (#433)
Browse files Browse the repository at this point in the history
  • Loading branch information
IgorTatarnikov authored Jun 3, 2024
1 parent 01f53e5 commit 093ba09
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions cellfinder/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@

# Set the Keras backend to torch
os.environ["KERAS_BACKEND"] = "torch"
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"

__license__ = "BSD-3-Clause"

Expand Down

0 comments on commit 093ba09

Please sign in to comment.