From 66fe11014892feb9120f4d0a3b0566ad1c77497b Mon Sep 17 00:00:00 2001
From: Jason Toffaletti <toffaletti@gmail.com>
Date: Wed, 31 Aug 2022 16:57:39 -0700
Subject: [PATCH] default full_prevision to True for mps device

---
 scripts/dream.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/scripts/dream.py b/scripts/dream.py
index 2911e8847a..1d6f4d2924 100755
--- a/scripts/dream.py
+++ b/scripts/dream.py
@@ -9,6 +9,7 @@ import sys
 import copy
 import warnings
 import time
+from ldm.dream.devices import choose_torch_device
 import ldm.dream.readline
 from ldm.dream.pngwriter import PngWriter, PromptFormatter
 from ldm.dream.server import DreamServer, ThreadingDreamServer
@@ -347,6 +348,8 @@ def create_argv_parser():
         dest='full_precision',
         action='store_true',
         help='Use slower full precision math for calculations',
+        # MPS only functions with full precision, see https://github.com/lstein/stable-diffusion/issues/237
+        default=choose_torch_device() == 'mps',
     )
     parser.add_argument(
         '-g',