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

Merge pull request #236 from Monadical-SAS/enhancements

Browse Source 
Upgrade to a bigger translation model
This commit is contained in:
projects-g
2023-09-20 20:07:15 +05:30
committed by GitHub
parent dc3d06038d 19dfb1d027
commit d7dbd6317c
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"