Patch TextTokens improper model_management usage

2023-11-14 15:19:19 -08:00
parent 75396f76af
commit 4e53775e65
1 changed files with 2 additions and 2 deletions
@@ -831,8 +831,8 @@ class TextTokens:
        self.tokens = {
            '[time]': str(time.time()).replace('.','_'),
            '[hostname]': socket.gethostname(),
-            '[cuda_device]': comfy.model_management.get_torch_device(),
-            '[cuda_name]': comfy.model_management.get_torch_device_name(),
+            '[cuda_device]': str(comfy.model_management.get_torch_device()),
+            '[cuda_name]': str(comfy.model_management.get_torch_device_name(device=comfy.model_management.get_torch_device())),
        }

        if '.' in self.tokens['[time]']: