Auto-deploy from GitHub: f28d8e4c0c69ce5da7e610a493fe42e9b8517bb8

src/spectttra/spectttra_trainer.py

@@ -128,7 +128,7 @@ def spectttra_predict(audio_tensor):
             melspec = torch.nn.functional.pad(melspec, (0, padding))
 
         if device.type == "cuda":
-            with torch.cuda.amp.autocast(enabled=True):
+            with torch.amp.autocast("cuda", enabled=True):
                 tokens = model(melspec)
                 pooled = tokens.mean(dim=1)
         else: