search

suno-ai / bark

🔊 Text-Prompted Generative Audio Model
MIT License
35.31k stars 4.15k forks source link

AttributeError: module 'torch.cuda' has no attribute 'is_bf16_supported' #40

Closed 5118Python closed 1 year ago

5118Python commented 1 year ago

Traceback (most recent call last): File "D:\5118\movielearning\testbark\test.py", line 1, in from bark import SAMPLE_RATE, generate_audio File "C:\ProgramData\Anaconda3\envs\movielearning\lib\site-packages\bark__init__.py", line 1, in from .api import generate_audio, text_to_semantic, semantic_to_waveform File "C:\ProgramData\Anaconda3\envs\movielearning\lib\site-packages\bark\api.py", line 5, in from .generation import codec_decode, generate_coarse, generate_fine, generate_text_semantic File "C:\ProgramData\Anaconda3\envs\movielearning\lib\site-packages\bark\generation.py", line 24, in torch.cuda.is_bf16_supported() AttributeError: module 'torch.cuda' has no attribute 'is_bf16_supported'

gkucsko commented 1 year ago

eeek, didn't know that can happen in some torch versions. will fix tyty

gkucsko commented 1 year ago

fixed in https://github.com/suno-ai/bark/commit/c3724301125c3ba4401c89958ca8612201282ec0