Commit 04cebec4 authored by liam's avatar liam
Browse files

rm opt config path default value and fix some config logic bug

parent dddc4203
...@@ -101,7 +101,7 @@ class ArgumentParser: ...@@ -101,7 +101,7 @@ class ArgumentParser:
parser.add_argument("--prompt_file", type=str, default=self.cfg.prompt_file) parser.add_argument("--prompt_file", type=str, default=self.cfg.prompt_file)
args = parser.parse_args() args = parser.parse_args()
if (args.model_dir is not None): if (args.model_dir is not None or args.model_path is not None):
if (args.model_path is not None): if (args.model_path is not None):
# if pass model_dir and model_path, we use model_path # if pass model_dir and model_path, we use model_path
args.model_dir = args.model_path args.model_dir = args.model_path
......
...@@ -95,7 +95,7 @@ class Config(metaclass=Singleton): ...@@ -95,7 +95,7 @@ class Config(metaclass=Singleton):
self.gguf_path: Optional[str] = self.model.get("gguf_path", None) self.gguf_path: Optional[str] = self.model.get("gguf_path", None)
# self.model_cache_lens = self.model.get("cache_lens") # self.model_cache_lens = self.model.get("cache_lens")
self.optimize_config_path: Optional[str] = self.model.get( self.optimize_config_path: Optional[str] = self.model.get(
"optimize_config_path", "./ktransformers/optimize/optimize_rules/DeepSeek-V2-Chat.yaml" "optimize_config_path", None
) )
self.paged = self.model.get("paged", True) self.paged = self.model.get("paged", True)
......
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