[CI] Fix black formatting on main (unblocks PR #21247 lint) (#24093)

Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
Kangyan-Zhou
2026-04-29 14:59:17 -07:00
committed by GitHub
co-authored by Claude Opus 4.7
parent 3d31ac2672
commit 6575aea128
3 changed files with 7 additions and 9 deletions
@@ -1003,9 +1003,7 @@ class ModelRunner(ModelRunnerKVCacheMixin):
sv = torch.empty(0, dtype=torch.uint8, device=t.device).set_(
t.untyped_storage()
)
if sv.data_ptr() not in {
v.data_ptr() for v in model_tensors.values()
}:
if sv.data_ptr() not in {v.data_ptr() for v in model_tensors.values()}:
model_tensors[f"{name}.__storage"] = sv
nixl_metadata = nixl_mgr.register_tensors(model_tensors)
+2 -5
View File
@@ -2408,9 +2408,7 @@ class RemoteInstanceModelLoader(BaseModelLoader):
# --- Transport-specific transfer ---
if transport == "nixl":
self._transfer_via_nixl(
model, nixl_mgr, source_worker, tp_rank
)
self._transfer_via_nixl(model, nixl_mgr, source_worker, tp_rank)
else:
self._transfer_via_transfer_engine(
model, transfer_engine, source_worker, tp_rank
@@ -2555,8 +2553,7 @@ class RemoteInstanceModelLoader(BaseModelLoader):
)
logger.info(
"ModelExpress [nixl]: transferred %d tensors, "
"%.2f GB in %.2fs",
"ModelExpress [nixl]: transferred %d tensors, " "%.2f GB in %.2fs",
matched,
total_bytes / 1e9,
duration,
+4 -1
View File
@@ -7109,7 +7109,10 @@ class ServerArgs:
if self.remote_instance_weight_loader_start_seed_via_transfer_engine:
return True
# ModelExpress source mode needs TransferEngine init only if transport is transfer_engine.
if self.modelexpress_source and self.modelexpress_transport == "transfer_engine":
if (
self.modelexpress_source
and self.modelexpress_transport == "transfer_engine"
):
return True
# Use TransferEngine as client backend.
elif (