Update requirements.txt

requirements.txt

@@ -1,19 +1,38 @@
-transformers
-accelerate
-python-slugify
-python-dotenv
+torch==2.6.0
+torchvision==0.21.0
+torchao==0.9.0
+safetensors
+git+https://github.com/huggingface/diffusers@363d1ab7e24c5ed6c190abb00df66d9edb74383b
+transformers==4.49.0
+lycoris-lora==1.8.3
+flatten_json
+pyyaml
+oyaml
+tensorboard
+kornia
+invisible-watermark
 einops
+accelerate
+toml
+albumentations==1.4.15
+albucore==0.0.16
+pydantic
+omegaconf
+k-diffusion
+open_clip_torch
 timm
-autotrain-advanced
-torch
-safetensors
-diffusers
-flash-attn
-ai-toolkit
-gradio
+prodigyopt
+controlnet_aux==0.0.7
+python-dotenv
+bitsandbytes
+hf_transfer
+lpips
+pytorch_fid
+optimum-quanto==0.2.4
+sentencepiece
 huggingface_hub
-omegaconf
-xformers
-
-bitsandbytes==0.45.0
-triton==2.0.0
+peft
+gradio
+python-slugify
+opencv-python
+pytorch-wavelets==1.3.0