Skip to content

Add Krea 2 segmented checkpointing support - #2947

Merged
bghira merged 1 commit into
agent/segmented-checkpointing-kolors-controlnetfrom
agent/segmented-checkpointing-krea2
Aug 3, 2026
Merged

Add Krea 2 segmented checkpointing support#2947
bghira merged 1 commit into
agent/segmented-checkpointing-kolors-controlnetfrom
agent/segmented-checkpointing-krea2

Conversation

@bghira

@bghira bghira commented Aug 2, 2026

Copy link
Copy Markdown
Owner

Summary

Splits the Krea 2 segmented checkpointing support model integration out of #2925.

  • applies the model-family checkpointing hooks and support flags
  • adds this family to the relevant safety-check allow-lists
  • keeps the shared runtime/docs in the base PR so this diff stays model-specific

Stack

Base branch: agent/segmented-checkpointing-kolors-controlnet

Validation

  • .venv/bin/python -m unittest tests.test_segmented_checkpointing_model_support -v
  • commit hooks: Black, isort, flake8, whitespace checks

@bghira
bghira force-pushed the agent/segmented-checkpointing-krea2 branch from 97a4917 to b426700 Compare August 3, 2026 09:02
@bghira
bghira marked this pull request as ready for review August 3, 2026 09:31
@bghira
bghira force-pushed the agent/segmented-checkpointing-krea2 branch from b426700 to b8c8349 Compare August 3, 2026 10:45
@bghira
bghira force-pushed the agent/segmented-checkpointing-krea2 branch from b8c8349 to 2c8596f Compare August 3, 2026 15:30
@bghira
bghira force-pushed the agent/segmented-checkpointing-krea2 branch from 2c8596f to da4a5f5 Compare August 3, 2026 15:32
@bghira
bghira force-pushed the agent/segmented-checkpointing-krea2 branch from da4a5f5 to 2d61c96 Compare August 3, 2026 15:33
@bghira
bghira force-pushed the agent/segmented-checkpointing-krea2 branch from 2d61c96 to 5a7eafa Compare August 3, 2026 15:35
@bghira
bghira merged commit 3504e55 into main Aug 3, 2026
@bghira
bghira deleted the agent/segmented-checkpointing-krea2 branch August 3, 2026 15:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant