refactor loading weights from remote instance coding format (#10941)
Signed-off-by: Anqi Shen <amy.saq@antgroup.com>
This commit is contained in:
@@ -91,7 +91,6 @@ class TpModelWorker:
|
||||
else server_args.speculative_draft_model_revision
|
||||
),
|
||||
is_draft_model=is_draft_worker,
|
||||
tp_rank=tp_rank,
|
||||
)
|
||||
|
||||
self.model_runner = ModelRunner(
|
||||
|
||||
Reference in New Issue
Block a user