Skip to content

Commit 80ff281

Browse files
RobotSailclaude
andcommitted
fix ruff format for CI version (0.15.5)
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent 9aef79d commit 80ff281

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/mini_trainer/setup_model_for_training.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -456,9 +456,7 @@ def wrap_fsdp2(model: torch.nn.Module) -> torch.nn.Module:
456456
# CheckpointError from tensor count mismatches.
457457
is_vlm_direct = hasattr(model, "model") and hasattr(model.model, "language_model")
458458
if is_vlm_direct:
459-
log_rank_0(
460-
f"🔄 [Phase 2] Skipping activation checkpointing for VLM ({len(layers)} blocks)"
461-
)
459+
log_rank_0(f"🔄 [Phase 2] Skipping activation checkpointing for VLM ({len(layers)} blocks)")
462460
else:
463461
log_rank_0(f"🔄 [Phase 2] Applying activation checkpointing to {len(layers)} blocks")
464462
for idx, block in enumerate(layers):

0 commit comments

Comments
 (0)