From ba9bcce65f98d94d2499e26ded2b2fdf897688db Mon Sep 17 00:00:00 2001 From: silveroxides Date: Mon, 13 Jul 2026 19:09:22 +0200 Subject: [PATCH] test: configure CPU mode for fusion tests --- tests-unit/comfy_extras_test/test_qwen_visual_fusion.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests-unit/comfy_extras_test/test_qwen_visual_fusion.py b/tests-unit/comfy_extras_test/test_qwen_visual_fusion.py index bed7fe9f4..a03739eba 100644 --- a/tests-unit/comfy_extras_test/test_qwen_visual_fusion.py +++ b/tests-unit/comfy_extras_test/test_qwen_visual_fusion.py @@ -1,5 +1,10 @@ import torch +from comfy.cli_args import args as cli_args + +if not torch.cuda.is_available(): + cli_args.cpu = True + from comfy_extras.nodes_qwen import TextEncodeQwenImageEditFusion, _flatten_images, _fuse_conditionings, _spatial_fusion_mask, _visual_token_span