selfhostyourtech/reflector
2
0
Fork 0
mirror of https://github.com/Monadical-SAS/reflector.git synced 2025-12-20 20:29:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Upgrade to a bigger translation model

Browse Source
This commit is contained in:
Gokul Mohanarangan
2023-09-20 20:02:52 +05:30
parent dc3d06038d
commit 19dfb1d027
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
server/gpu/modal/reflector_transcriber.py
View File

@@ -6,7 +6,6 @@ Reflector GPU backend - transcriber
import os
import tempfile
import modal
from modal import Image, Secret, Stub, asgi_app, method
from pydantic import BaseModel
@@ -16,7 +15,7 @@ WHISPER_COMPUTE_TYPE: str = "float16"
WHISPER_NUM_WORKERS: int = 1
# Translation Model
TRANSLATION_MODEL = "facebook/m2m100_418M"
TRANSLATION_MODEL = "facebook/m2m100_1.2B"
IMAGE_MODEL_DIR = "/root/transcription_models"