Skip to content

Conversation

hipsterusername
Copy link
Member

Summary

Updates to model probe to handle PEFT models output with base_model.model

Related Issues / Discussions

Handles conversion issues with native pre-conversion default model format out of invoke training

QA Instructions

  • Import a Flux model trained using invoke-training that isn't converted to kohya keys.

Merge Plan

  • Merge at will.

Checklist

  • The PR has a short but descriptive title, suitable for a changelog
  • Tests added / updated (if applicable)
  • Documentation added / updated (if applicable)
  • Updated What's New copy (if doing a release after this PR)

@github-actions github-actions bot added python PRs that change python files backend PRs that change backend files labels Aug 17, 2025
@github-actions github-actions bot added the python-tests PRs that change python tests label Aug 18, 2025
@maryhipp maryhipp merged commit af58a75 into main Aug 18, 2025
12 checks passed
@maryhipp maryhipp deleted the peft-base-model-flux branch August 18, 2025 13:14
maryhipp pushed a commit that referenced this pull request Aug 18, 2025
* Support PEFT Loras with Base_Model.model prefix

* update tests

* ruff

* fix python complaints

* update kes

* format keys

* remove unneeded test
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
backend PRs that change backend files python PRs that change python files python-tests PRs that change python tests
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants