From 045464d9e13ad2d0f3a2171d052a34b6ba946535 Mon Sep 17 00:00:00 2001 From: Danny Semiat Date: Sun, 17 Mar 2024 15:29:08 +0200 Subject: [PATCH] Fixed const serialization path call to bridge Update utils.py --- examples/text-generation/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/text-generation/utils.py b/examples/text-generation/utils.py index 861d027a26..0df71de329 100644 --- a/examples/text-generation/utils.py +++ b/examples/text-generation/utils.py @@ -111,7 +111,7 @@ def setup_const_serialization(const_serialization_path): os.makedirs(const_serialization_path) from habana_frameworks.torch.hpu import enable_const_section_serialization print("Serializing const params to {}".format(const_serialization_path)) - enable_const_section_serialization(const_serialization_path, False, True) + enable_const_section_serialization(const_serialization_path, True) def setup_env(args): # Will error if the minimal version of Transformers is not installed. Remove at your own risks.