FutureWarning: `torch.cuda.amp.autocast(args...)` is deprecated on 0.0.16

Stability-AI / stable-audio-tools

Generative models for conditional audio generation

MIT License

2.72k stars 258 forks source link

FutureWarning: `torch.cuda.amp.autocast(args...)` is deprecated on 0.0.16 #143

Open carlthome opened 1 month ago

carlthome commented 1 month ago

Getting deprecation warnings when running stable-audio-tools==0.0.16

.venv/lib/python3.10/site-packages/stable_audio_tools/models/conditioners.py:314: FutureWarning: `torch.cuda.amp.autocast(args...)` is deprecated. Please use `torch.amp.autocast('cuda', args...)` instead.