Skip to content

GitLab

Unverified Commit 0f054fe4 authored by UnicornChan's avatar UnicornChan Committed by GitHub
Browse files

Merge pull request #52 from UnicornChan/fix-bug-load-config 

Fix: None for load config
parents 1f85db3d b9f0819a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ktransformers/server/backend/interfaces/ktransformers.py
View file @ 0f054fe4
...@@ -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