load on cpu

tasks/audio.py

@@ -77,7 +77,7 @@ async def evaluate_audio(request: AudioEvaluationRequest):
 
     model = CNNKan(model_args, conformer_args, kan_args.get_dict())
     model = model.to(device)
-    state_dict = torch.load(data_args.checkpoint_path)
+    state_dict = torch.load(data_args.checkpoint_path, map_location=torch.device('cpu'))
     new_state_dict = OrderedDict()
     for key, value in state_dict.items():
         if key.startswith('module.'):