enlyth / sd-webui-riffusion

Riffusion extension for AUTOMATIC1111's SD Web UI

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

possible fix for index error?

wedgeewoo opened this issue · comments

I was getting this error when trying to convert pngs in a directory :

File "A:\Auto1111-SD-Repo\stable-diffusion-webui\extensions\sd-webui-riffusion\scripts\riffusion.py", line 217, in spectrogram_from_image
data = data[::-1, :, 0]
IndexError: too many indices for array: array is 2-dimensional, but 3 were indexed

I just changed line 217 to : data = data[::-1, :]

The feature seems to work for me now

nvm it just broke the image generation

File "A:\Auto1111-SD-Repo\stable-diffusion-webui\venv\lib\site-packages\torch\functional.py", line 630, in stft
input = F.pad(input.view(extended_shape), [pad, pad], pad_mode)
RuntimeError: Argument #4: Padding size should be less than the corresponding input dimension, but got: padding (8820, 8820) at dimension 2 of input [1, 512, 882]