wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-08 12:32:32 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

improve tests

Browse Source
This commit is contained in:
doctorpangloss 2025-12-09 19:51:55 -08:00
parent f294c5d401
commit e9d5210fcc
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy_extras/nodes/nodes_audio.py
View File

@ -34,7 +34,7 @@ class EmptyLatentAudio(IO.ComfyNode):
) )
@classmethod @classmethod
def execute(cls, seconds, batch_size) -> IO.NodeOutput: def execute(cls, seconds=47.6, batch_size=1) -> IO.NodeOutput:
length = round((seconds * 44100 / 2048) / 2) * 2 length = round((seconds * 44100 / 2048) / 2) * 2
latent = torch.zeros([batch_size, 64, length], device=comfy.model_management.intermediate_device()) latent = torch.zeros([batch_size, 64, length], device=comfy.model_management.intermediate_device())
return IO.NodeOutput({"samples": latent, "type": "audio"}) return IO.NodeOutput({"samples": latent, "type": "audio"})

5
tests/inference/test_workflows.py
View File

@ -60,7 +60,12 @@ def _generate_config_params():
@pytest.fixture(scope="function", autouse=False, params=_generate_config_params()) @pytest.fixture(scope="function", autouse=False, params=_generate_config_params())
async def client(tmp_path_factory, request) -> AsyncGenerator[Any, Any]: async def client(tmp_path_factory, request) -> AsyncGenerator[Any, Any]:
config = default_configuration() config = default_configuration()
# this should help things go a little faster
config.disable_all_custom_nodes = True
# this enables compilation
config.disable_pinned_memory = True
config.update(request.param) config.update(request.param)
# use ProcessPoolExecutor to respect various config settings
async with Comfy(configuration=config, executor=ProcessPoolExecutor(max_workers=1)) as client: async with Comfy(configuration=config, executor=ProcessPoolExecutor(max_workers=1)) as client:
yield client yield client