haoheliu / versatile_audio_super_resolution

Versatile audio super resolution (any -> 48kHz) with AudioSR.
MIT License
1.1k stars 107 forks source link

CUDA out of memory? #65

Open KLmining opened 3 weeks ago

KLmining commented 3 weeks ago

My graphics card is RTX 3060ti 8gb, I cut the audio to 0.5 seconds, but the problem still occurs.What should I do?

clip-1

djethino commented 1 day ago

I made a little script for windows to auto split the sound and to re-concatenate it after the enhancement : requirement:

AudioSR-Split.bat

@echo off
:: customisable variables
SET model_name=basic
SET segment_time=5.12
SET seed=27
SET ddim_steps=50
SET outputfolder=.\output

:: parameters
if [%1]==[] goto :help
if [%2]==[] goto :help
SET file_source="%1"
SET file_target="%2"

color 12
echo ===================================================================
echo Starting process audio enhancer for %file_source% to %file_target%
echo ===================================================================
color 01

ffmpeg -i %file_source% -f segment -segment_time %segment_time% -c copy "%outputfolder%\out_%%05d.wav" -y >> %outputfolder%\output.log

color 12
echo ===============================
echo Preparing files list: wavlistaudiosr.txt
echo ===============================
color 01

for %%f in (%outputfolder%\out_*.wav) do (
    @echo off
    echo %outputfolder%\%%~nf%%~xf >> %outputfolder%\wavlistaudiosr.txt
)

color 12
echo ===============================
echo Processing enhancement
echo ===============================
color 35

call audiosr -il %outputfolder%\wavlistaudiosr.txt --seed %seed% --model_name %model_name% --ddim_steps %ddim_steps% -s %outputfolder% >> %outputfolder%\output.log

set c=1
for /f "delims=" %%a in ('dir %outputfolder% /b/ad/o-d') do if %c%==1 (set lastfolder=%%a
                                                                    goto:next)
:next
color 12
echo ===============================
echo Preparing files list: wavlistffmpeg.txt
echo ===============================
color 01

set outlast=%outputfolder%\%lastfolder%

for %%f in (%outlast%\*.wav) do (
    @echo off
    echo file '%lastfolder%\%%~nf%%~xf' >> %outputfolder%\wavlistffmpeg.txt
)

color 12
echo ===============================
echo Processing concatenation
echo ===============================
color 01

ffmpeg -f concat -safe 0 -i %outputfolder%\wavlistffmpeg.txt -c:a libmp3lame %file_target% -y >> %outputfolder%\output.log

del %outputfolder%\wavlistaudiosr.txt
del %outputfolder%\wavlistffmpeg.txt
del %outputfolder%\output.log
del %outputfolder%\out_*.wav
rd /s /q %outlast%

color 12
echo ===============================
echo Exported file: %file_target%
echo ===============================
color 01

goto :end

:help
echo Usage:
echo %~nx0 source target

:end
color 07

Usage: .\AudioSR-Split.bat source target