enlyth / sd-webui-riffusion

Riffusion extension for AUTOMATIC1111's SD Web UI
MIT License
195 stars 23 forks source link

Images to error, batch problem #31

Open imago10 opened 1 year ago

imago10 commented 1 year ago

Hi when I try to convert my spectograms from folder I got this error, any idea how to fix that?

Found 48 images in C:\RIFF\fron, pattern .jpg, .png Traceback (most recent call last): File "C:\WEBUI_130\stable-diffusion-webui\venv\lib\site-packages\gradio\routes.py", line 414, in run_predict output = await app.get_blocks().process_api( File "C:\WEBUI_130\stable-diffusion-webui\venv\lib\site-packages\gradio\blocks.py", line 1323, in process_api result = await self.call_function( File "C:\WEBUI_130\stable-diffusion-webui\venv\lib\site-packages\gradio\blocks.py", line 1051, in call_function prediction = await anyio.to_thread.run_sync( File "C:\WEBUI_130\stable-diffusion-webui\venv\lib\site-packages\anyio\to_thread.py", line 33, in run_sync return await get_asynclib().run_sync_in_worker_thread( File "C:\WEBUI_130\stable-diffusion-webui\venv\lib\site-packages\anyio_backends_asyncio.py", line 877, in run_sync_in_worker_thread return await future File "C:\WEBUI_130\stable-diffusion-webui\venv\lib\site-packages\anyio_backends_asyncio.py", line 807, in run result = context.run(func, *args) File "C:\WEBUI_130\stable-diffusion-webui\extensions\sd-webui-riffusion\scripts\riffusion.py", line 364, in convert_audio output_files.append(convert_audio_image(image, image_file, image_dir, width)) File "C:\WEBUI_130\stable-diffusion-webui\extensions\sd-webui-riffusion\scripts\riffusion.py", line 333, in convert_audio_image wav_bytes, duration_s = riffusion.wav_bytes_from_spectrogram_image(image_file) File "C:\WEBUI_130\stable-diffusion-webui\extensions\sd-webui-riffusion\scripts\riffusion.py", line 163, in wav_bytes_from_spectrogram_image Sxx = self.spectrogram_from_image( File "C:\WEBUI_130\stable-diffusion-webui\extensions\sd-webui-riffusion\scripts\riffusion.py", line 218, in spectrogram_from_image data = data[::-1, :, 0] IndexError: too many indices for array: array is 2-dimensional, but 3 were indexed