ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-01 11:41:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added experimental wavelet loss

Browse Source
This commit is contained in:
Jaret Burkett
2025-03-26 18:11:23 -06:00
parent c101f07834
commit ce4c5291a0
4 changed files with 45 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
requirements.txt
View File

@@ -34,4 +34,5 @@ huggingface_hub
peft
gradio
python-slugify
opencv-python
opencv-python
pytorch-wavelets==1.3.0