Commit b9f0819a authored by chenxl's avatar chenxl
Browse files

None for load config

parent 1f85db3d
...@@ -25,7 +25,7 @@ class KTransformersInterface(TransformersInterface): ...@@ -25,7 +25,7 @@ class KTransformersInterface(TransformersInterface):
with torch.device("meta"): with torch.device("meta"):
self.model=custom_models[config.architectures[0]](config) self.model=custom_models[config.architectures[0]](config)
if default_args.optimize_config_path is not None: if default_args.optimize_config_path is None:
optimize_rule_path = default_optimize_rules[config.architectures[0]] optimize_rule_path = default_optimize_rules[config.architectures[0]]
else: else:
optimize_rule_path = args.optimize_config_path optimize_rule_path = args.optimize_config_path
......
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