From 887018b0f2fb4dbcf5ec1c197ffc576f456f2e1f Mon Sep 17 00:00:00 2001 From: Zhao Changmin Date: Wed, 30 Aug 2023 10:32:57 +0800 Subject: [PATCH] Update ut save&load (#8847) Co-authored-by: leonardozcm --- python/llm/test/convert/test_convert_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/llm/test/convert/test_convert_model.py b/python/llm/test/convert/test_convert_model.py index ddaaad91..e8cc3079 100644 --- a/python/llm/test/convert/test_convert_model.py +++ b/python/llm/test/convert/test_convert_model.py @@ -67,7 +67,7 @@ class TestConvertModel(TestCase): def test_transformer_convert_llama(self): with tempfile.TemporaryDirectory(dir=output_dir) as tempdir: model = AutoModelForCausalLM.from_pretrained(llama_model_path, load_in_4bit=True) - model.save_pretrained(tempdir) + model.save_low_bit(tempdir) newModel = AutoModelForCausalLM.load_low_bit(tempdir) assert newModel is not None