Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #2296 from vladmandic/master

Browse Source 
update dml
This commit is contained in:
Vladimir Mandic
2023-10-08 07:26:55 -04:00
committed by GitHub
parent 65dfed93a1 e5f8b7f0a4
commit 564d04d9f4
4 changed files with 49 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dml/__init__.py
+6
View File
@@ -10,6 +10,7 @@ if platform.system() == "Windows":
memory_providers.append("Performance Counter")
default_memory_provider = "Performance Counter"
do_nothing = lambda: None # pylint: disable=unnecessary-lambda-assignment
do_nothing_with_self = lambda self: None # pylint: disable=unnecessary-lambda-assignment
def _set_memory_provider():
from modules.shared import opts, cmd_opts
@@ -66,7 +67,12 @@ def directml_do_hijack():
import modules.dml.hijack # pylint: disable=unused-import
from modules.devices import device
CondFunc('torch.Generator',
lambda orig_func, device: orig_func("cpu"),
lambda orig_func, device: True)
if not torch.dml.has_float64_support(device):
torch.Tensor.__str__ = do_nothing_with_self
CondFunc('torch.from_numpy',
lambda orig_func, *args, **kwargs: orig_func(args[0].astype('float32')),
lambda *args, **kwargs: args[1].dtype == float)