diff --git a/requirements.txt b/requirements.txt
index de39f588..52c3b8c7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -43,4 +43,5 @@ rich==13.7.0
 # for T5XXL tokenizer (SD3/FLUX)
 sentencepiece==0.2.0
 # for kohya_ss library
+pytorch-optimizer
 -e .