wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-20 05:18:15 +08:00
Code Issues Actions 25 Packages Projects Releases Wiki Activity

test: configure CPU mode for fusion tests

Browse Source
This commit is contained in:
silveroxides 2026-07-13 19:09:22 +02:00
parent 658e711356
commit ba9bcce65f
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests-unit/comfy_extras_test/test_qwen_visual_fusion.py
View File

@ -1,5 +1,10 @@
import torch 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 from comfy_extras.nodes_qwen import TextEncodeQwenImageEditFusion, _flatten_images, _fuse_conditionings, _spatial_fusion_mask, _visual_token_span