Skip to content

Commit 65b7ec2

Browse files
authored
fix vlm gguf ut (#895)
Signed-off-by: n1ck-guo <heng.guo@intel.com>
1 parent 7beac1b commit 65b7ec2

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

test/test_cpu/test_gguf_format.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -329,7 +329,6 @@ def test_vlm_gguf(self):
329329
autoround.quantize_and_save(output_dir=quantized_model_path, format="gguf:q4_0")
330330
self.assertTrue("mmproj-model.gguf" in os.listdir("./saved"))
331331
for file_name in os.listdir(quantized_model_path):
332-
file_name = os.listdir(quantized_model_path)[0]
333332
file_size = os.path.getsize(os.path.join(quantized_model_path, file_name)) / 1024**2
334333
if file_name == "mmproj-model.gguf":
335334
self.assertAlmostEqual(file_size, 2535, delta=1.0)

0 commit comments

Comments
 (0)