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

Suggest the right module

Browse Source
This commit is contained in:
rockerBOO
2025-04-08 04:13:37 -04:00
parent 837231a5c7
commit 64422ff4a0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/custom_train_functions.py
View File

@@ -588,7 +588,7 @@ class WaveletLoss(torch.nn.Module):
self.hl_weight2 = 0.01
self.hh_weight2 = 0.05
assert pywt.wavedec2 is not None, "PyWavelet module not available. Please install `pip install PyWavelet`"
assert pywt.wavedec2 is not None, "PyWavelets module not available. Please install `pip install PyWavelets`"
# Create GPU filters from wavelet
wav = pywt.Wavelet(wavelet)
self.register_buffer('dec_lo', torch.Tensor(wav.dec_lo).to(device))