test: configure CPU mode for fusion tests

2026-07-13 19:09:22 +02:00 · 2026-07-13 19:09:22 +02:00 · ba9bcce65f
parent 658e711356
commit ba9bcce65f
1 changed files with 5 additions and 0 deletions
--- 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