Skip to content

Conversation

kohya-ss
Copy link
Owner

  • Upgrade PyTorch >= 2.6.0
  • Upgrade the dependent libraries to versions that are confirmed to work with Musubi Tuner.
  • Resolve all possible warnings.

@kohya-ss
Copy link
Owner Author

pytorch-lightning is finally removed... Also gradio and altair doesn't seem to be needed anymore.

@rockerBOO
Copy link
Contributor

Can add python 2.6.0 to the github workflow too. Maybe we keep both 2.4.0 and 2.6.0 for now.

@kohya-ss
Copy link
Owner Author

Can add python 2.6.0 to the github workflow too. Maybe we keep both 2.4.0 and 2.6.0 for now.

That's right. I believe the project will continue to work with PyTorch 2.4.0. I'll look into how to change the flow, but it might be a good idea to merge this PR first.

@rockerBOO
Copy link
Contributor

Would need to change this line, to add 2.6.0

pytorch-version: ["2.4.0"] # PyTorch versions to test

@kohya-ss
Copy link
Owner Author

The scripts updated to remove the warnings seem to work with previous library versions as well, so users can upgrade their environments at any time.

There seems to be no problem merging this PR.

@kohya-ss kohya-ss merged commit 8cadec6 into sd3 Aug 27, 2025
3 checks passed
@kohya-ss kohya-ss deleted the update-libraries branch August 27, 2025 23:42
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.

2 participants