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

test: restore CLI state after node import

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

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

@ -2,10 +2,14 @@ import torch
from comfy.cli_args import args as cli_args
prior_cpu = cli_args.cpu
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
try:
from comfy_extras.nodes_qwen import TextEncodeQwenImageEditFusion, _flatten_images, _fuse_conditionings, _spatial_fusion_mask, _visual_token_span
finally:
cli_args.cpu = prior_cpu
def _tokens(image_position=1, suffix=1):