Skip to content

Commit

Permalink
fix spurious
Browse files Browse the repository at this point in the history
  • Loading branch information
robertgshaw2-neuralmagic committed Oct 21, 2024
1 parent bb2fbe1 commit 14ccb90
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion vllm/model_executor/model_loader/loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -411,7 +411,6 @@ def load_model(self, *, model_config: ModelConfig,
# parameters onto device for processing and back off after.
with device_loading_context(module, target_device):
quant_method.process_weights_after_loading(module)

return model.eval()


Expand Down Expand Up @@ -1147,6 +1146,7 @@ def load_model(self, *, model_config: ModelConfig,
lora_config, cache_config)

self._load_weights(model_config, model)

return model.eval()


Expand Down

0 comments on commit 14ccb90

Please sign in to comment.