Commit 034aae03 authored by zhuwenwen's avatar zhuwenwen
Browse files

update gen_model_json

parent 1c7f32cd
...@@ -2232,7 +2232,7 @@ class W8a8GetCacheJSON: ...@@ -2232,7 +2232,7 @@ class W8a8GetCacheJSON:
#析构函数,最后会生成model.json的配置文件 #析构函数,最后会生成model.json的配置文件
def gen_model_json(self,E:Optional[int]=0,block_size:Optional[list]=None): def gen_model_json(self,E:Optional[int]=0,block_size:Optional[list]=None):
json_dir = os.getenv('LMSLIM_TUNING_JSON', "None") json_dir = os.getenv('LMSLIM_TUNING_JSON', "None")
if json_dir is not "None" and os.path.exists(json_dir): if json_dir != "None" and os.path.exists(json_dir):
#生成模型配置文件 #生成模型配置文件
# logger.info("model_tuning.json is at LMSLIM_TUNING_JSON:%s", json_dir) # logger.info("model_tuning.json is at LMSLIM_TUNING_JSON:%s", json_dir)
config = { config = {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment