🚧 [WIP/TEST] Model : CUDAでの処理実装 part 1

2024-10-18 16:16:42 +09:00
parent 147b5cc396
commit 5ba595cefe
2 changed files with 3 additions and 3 deletions
--- a/src-python/models/transcription/transcription_whisper.py
+++ b/src-python/models/transcription/transcription_whisper.py
@@ -51,7 +51,7 @@ def checkWhisperWeight(root, weight_type):
    try:
        WhisperModel(
            path,
-            device="cpu",
+            device="cuda",
            device_index=0,
            compute_type="int8",
            cpu_threads=4,
@@ -79,7 +79,7 @@ def getWhisperModel(root, weight_type):
    path = os_path.join(root, "weights", "whisper", weight_type)
    return WhisperModel(
        path,
-        device="cpu",
+        device="cuda",
        device_index=0,
        compute_type="int8",
        cpu_threads=4,