Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
brosoul committed Nov 8, 2024
1 parent 1f27913 commit b167008
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions vllm/model_executor/model_loader/loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -1197,8 +1197,7 @@ def load_model(self, *, model_config: ModelConfig,
lora_config, cache_config,
scheduler_config)

model.load_weights(
self.stream_model.get_weights_iterator("cpu"))
model.load_weights(self.stream_model.get_weights_iterator("cpu"))

return model.eval()

Expand Down

0 comments on commit b167008

Please sign in to comment.