[Fix] Call init_group every time to resolve the LoRA training issue when Dropout is enabled in Kohya-ss

[kozistr]

Problem (Why?)

Fixes #462

Solution (What/How?)

• call init_group() every time

Other changes (bug fixes, small refactors)

N/A

Notes

N/A

Checklist

• Make sure to run make format before commit
• My code adheres to the style guidelines of this project (make check shows no errors)
• Both new and existing unit tests pass successfully on my local environment by running make test
• I have made the necessary changes to the documentation

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 100.00%. Comparing base (1b1313b) to head (baec6fa).
⚠️ Report is 1 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff            @@
##              main      #463   +/-   ##
=========================================
  Coverage   100.00%   100.00%           
=========================================
  Files          120       120           
  Lines        11010     11028   +18     
=========================================
+ Hits         11010     11028   +18     

Flag	Coverage Δ
3.10	100.00% <100.00%> (ø)
3.11	100.00% <100.00%> (ø)
3.12	100.00% <100.00%> (ø)
3.13	100.00% <100.00%> (ø)
3.14	100.00% <100.00%> (ø)
3.8	99.99% <100.00%> (+<0.01%)	⬆️
3.9	99.99% <100.00%> (+<0.01%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.