add support for GPU (cuda)

2023-10-27 10:23:08 +02:00
parent f4484884cc
commit 5570a745f8
7 changed files with 170 additions and 22 deletions
--- a/backend/api/handlers.go
+++ b/backend/api/handlers.go
@@ -87,8 +87,11 @@ func (s *Server) handlePostTranscription(c *fiber.Ctx) error {
 	transcription.Status = models.TranscriptionStatusPending
 	transcription.Task = "transcribe"
 	transcription.SourceUrl = c.FormValue("sourceUrl")
-	// TODO: Implement other device types
-	transcription.Device = "cpu"
+	transcription.Device = c.FormValue("device")
+	if transcription.Device != "cpu" && transcription.Device != "cuda" {
+		log.Warn().Msgf("Device %v not supported, using cpu", transcription.Device)
+		transcription.Device = "cpu"
+	}

 	log.Debug().Msgf("Transcription: %+v", transcription)
 	// Save transcription to database