From 193eae246cb91ff25fe7ad696af902ee736ef510 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Thu, 6 Jul 2023 22:02:23 +0000 Subject: [PATCH] Update test. --- server/tests/utils/test_convert.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/server/tests/utils/test_convert.py b/server/tests/utils/test_convert.py index 7dfe6a1e..ba6c5702 100644 --- a/server/tests/utils/test_convert.py +++ b/server/tests/utils/test_convert.py @@ -14,7 +14,7 @@ def test_convert_files(): local_st_files = [ p.parent / f"{p.stem.lstrip('pytorch_')}.safetensors" for p in local_pt_files ] - convert_files(local_pt_files, local_st_files) + convert_files(local_pt_files, local_st_files, discard_names=[]) found_st_files = weight_files(model_id)