# base requirements
numpy==1.26.4
opencv-python==4.10.0.84
pillow==11.0.0
tqdm==4.66.6
PyYAML==6.0.2
huggingface-hub==0.26.2
scipy==1.14.1; sys_platform != 'win32'
matplotlib==3.9.2

# pytorch
accelerate==1.0.1
safetensors==0.4.5
tensorboard==2.18.0
pytorch-lightning==2.4.0

# stable diffusion
-e git+https://github.com/huggingface/diffusers.git@55ac1db#egg=diffusers
transformers==4.47.0
sentencepiece==0.2.0 # transitive dependency of transformers for tokenizer loading
omegaconf==2.3.0 # needed to load stable diffusion from single ckpt files
invisible-watermark==0.2.0 # needed for the SDXL pipeline

# other models
pooch==1.8.2
open-clip-torch==2.28.0

# data loader
-e git+https://github.com/Nerogar/mgds.git@e6bd96b#egg=mgds

# optimizers
dadaptation==3.2 # dadaptation optimizers
lion-pytorch==0.2.2 # lion optimizer
prodigyopt==1.1.1 # prodigy optimizer
schedulefree==1.3.0 # schedule-free optimizers
pytorch_optimizer==3.3.0 # pytorch optimizers

# Profiling
scalene==1.5.45

# ui
customtkinter==5.2.2
