"vscode:/vscode.git/clone" did not exist on "ab7a78e8f11eec914653e01ee497d57d7503bd9d"
Unverified Commit bd619616 authored by Lianmin Zheng's avatar Lianmin Zheng Committed by GitHub
Browse files

Small fix for the order of apply_torchao_config (#2495)

parent 56198b45
......@@ -157,6 +157,11 @@ class ModelRunner:
self.sampler = Sampler()
self.load_model()
# Apply torchao quantization
apply_torchao_config_to_model(
self.model, global_server_args_dict["torchao_config"]
)
# Apply torch TP if the model supports it
supports_torch_tp = getattr(self.model, "supports_torch_tp", False)
if self.tp_size > 1 and supports_torch_tp:
......@@ -165,10 +170,6 @@ class ModelRunner:
else:
self.torch_tp_applied = False
apply_torchao_config_to_model(
self.model, global_server_args_dict["torchao_config"]
)
# Init memory pool and attention backends
if server_args.lora_paths is not None:
self.init_lora_manager()
......
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