فهرست منبع

[1.0] Update to float32 to support my GPU

Penta 9 ماه پیش
والد
کامیت
076f25e321
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  1. 1 1
      app.py

+ 1 - 1
app.py

@@ -8,7 +8,7 @@ app = FastAPI()
 model = WhisperModel(
     "small",
     device="cuda",
-    compute_type="float16"
+    compute_type="float32"
 )
 
 @app.post("/transcribe")