From 4e53775e650a7e2d2d1d73056bb914d7edc57f69 Mon Sep 17 00:00:00 2001 From: Jordan Thompson Date: Tue, 14 Nov 2023 15:19:19 -0800 Subject: [PATCH] Patch TextTokens improper model_management usage --- WAS_Node_Suite.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/WAS_Node_Suite.py b/WAS_Node_Suite.py index 0db7094..ed4a3bc 100644 --- a/WAS_Node_Suite.py +++ b/WAS_Node_Suite.py @@ -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]']: