From 6f7fe5e7cc03a3449201a9da3c35f50a539bb86e Mon Sep 17 00:00:00 2001
From: Seunghoon Lee <lshqqytiger@naver.com>
Date: Fri, 20 Sep 2024 10:09:00 +0900
Subject: [PATCH] fix directml generator.

---
 modules/dml/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/modules/dml/__init__.py b/modules/dml/__init__.py
index b289e5d2d..c0ddbc1a1 100644
--- a/modules/dml/__init__.py
+++ b/modules/dml/__init__.py
@@ -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