Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

default full_prevision to True for mps device

Browse Source
This commit is contained in:
Jason Toffaletti
2022-08-31 16:57:39 -07:00
parent bf50ab9dd6
commit 66fe110148
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/dream.py
View File

@ -9,6 +9,7 @@ import sys
import copy import copy
import warnings import warnings
import time import time
from ldm.dream.devices import choose_torch_device
import ldm.dream.readline import ldm.dream.readline
from ldm.dream.pngwriter import PngWriter, PromptFormatter from ldm.dream.pngwriter import PngWriter, PromptFormatter
from ldm.dream.server import DreamServer, ThreadingDreamServer from ldm.dream.server import DreamServer, ThreadingDreamServer
@ -347,6 +348,8 @@ def create_argv_parser():
dest='full_precision', dest='full_precision',
action='store_true', action='store_true',
help='Use slower full precision math for calculations', 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( parser.add_argument(
'-g', '-g',