diff --git a/ktransformers/server/backend/interfaces/ktransformers.py b/ktransformers/server/backend/interfaces/ktransformers.py index f896a90..0e5a8c8 100644 --- a/ktransformers/server/backend/interfaces/ktransformers.py +++ b/ktransformers/server/backend/interfaces/ktransformers.py @@ -25,7 +25,7 @@ class KTransformersInterface(TransformersInterface): with torch.device("meta"): 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]] else: optimize_rule_path = args.optimize_config_path