Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix directml generator.

Browse Source
This commit is contained in:
Seunghoon Lee
2024-09-20 10:09:00 +09:00
parent 4acf3c00a2
commit 6f7fe5e7cc
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dml/__init__.py
+2 -2
View File
@@ -72,8 +72,8 @@ def directml_do_hijack():
from modules.devices import device
CondFunc('torch.Generator',
lambda orig_func, device: orig_func("cpu"),
lambda orig_func, device: True)
lambda orig_func, device = None: orig_func("cpu"),
lambda orig_func, device = None: True)
if not torch.dml.has_float64_support(device):
torch.Tensor.__str__ = do_nothing_with_self