Update tasks/audio.py

tasks/audio.py

@@ -100,7 +100,7 @@ async def evaluate_audio(request: AudioEvaluationRequest):
     #--------------------------------------------------------------------------------------------   
     predictions = []
     with torch.no_grad():
-        with autocast():  
+        with autocast(device_type='cpu'):  
             for data, target in test_loader:
                 outputs = scripted_model(data)
                 _, predicted = torch.max(outputs, 1)