VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-16 17:02:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1985 from gesen2egee/pytorch-optimizer

Browse Source 
Support pytorch_optimizer
This commit is contained in:
Kohya S.
2025-03-20 22:01:03 +09:00
committed by GitHub
parent 75933d70a1 5b210ad717
commit 936d333ff4
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
requirements.txt
View File

@@ -7,7 +7,6 @@ opencv-python==4.8.1.78
einops==0.7.0
pytorch-lightning==1.9.0
bitsandbytes==0.44.0
prodigyopt==1.0
lion-pytorch==0.0.6
schedulefree==1.4
tensorboard
@@ -43,4 +42,7 @@ rich==13.7.0
# for T5XXL tokenizer (SD3/FLUX)
sentencepiece==0.2.0
# for kohya_ss library
pytorch-optimizer
prodigy-plus-schedule-free==1.9.0
prodigyopt==1.1.2
-e .