wsippel / bark_tts

Oobabooga extension for Bark TTS
GNU Affero General Public License v3.0
110 stars 14 forks source link

PYTORCH_CUDA_ALLOC_CONF #12

Open AssetDK opened 1 year ago

AssetDK commented 1 year ago

Hi - thanks for some really good instructions for installation. Sadly this was hard to find, but I finally did.

Perhaps you can help... while BARK_TT seems to load ok, I get this CUDA error when loading - saying that I could add some PYTORCH_CUDA_ALLOC_CONF paramter and I tried a few things, but it does not work... I tried this:

Add this statement for the start of @rem config part of start_windows.bat

Added run_cmd("set 'PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512'" ) # both with and without environment=True

Like for many of the readme files and instructions it is not really clear where to add these paramters/commands? And it is not easy to know if they are actually set correctly but I tired to run this: echo %PYTORCH_CUDA_ALLOC_CONF% and is yields the correct value...