wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-21 23:41:28 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Fixes to tests and progress bars in xet

Browse Source
This commit is contained in:
doctorpangloss
2025-08-01 17:26:30 -07:00
parent 87bed08124
commit 59bc6afa7b
19 changed files with 104 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/quantization/test_torchao.py
+3 -3
View File
@@ -39,21 +39,21 @@ def model_patcher_obj(request) -> ModelPatcher:
model_management.soft_empty_cache(force=True)
@pytest.mark.forked
# @pytest.mark.forked
@pytest.mark.skipif(not has_torchao, reason="torchao not installed")
@pytest.mark.skipif(True, reason="wip")
async def test_unit_torchao(model_patcher_obj):
quantize_(model_patcher_obj.diffusion_model, int8_dynamic_activation_int8_weight(), device=model_management.get_torch_device())
@pytest.mark.forked
# @pytest.mark.forked
@pytest.mark.parametrize("strategy", ["torchao", "torchao-autoquant"])
@pytest.mark.skipif(True, reason="wip")
async def test_torchao_node(model_patcher_obj, strategy):
QuantizeModel().execute(model_patcher_obj, strategy=strategy)
@pytest.mark.forked
# @pytest.mark.forked
@pytest.mark.skipif(True, reason="wip")
async def test_tensorrt(model_patcher_obj):
STATIC_TRT_MODEL_CONVERSION().convert(model_patcher_obj, "test", 1, 1024, 1024, 1, 14)