From 895ba36cb93882cfae69b260d25fcd09896307d5 Mon Sep 17 00:00:00 2001 From: Mathieu Virbel Date: Tue, 1 Oct 2024 16:39:24 +0200 Subject: [PATCH] fix: modal upgrade (#421) --- server/gpu/modal_deployments/reflector_diarizer.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/server/gpu/modal_deployments/reflector_diarizer.py b/server/gpu/modal_deployments/reflector_diarizer.py index db46b83c..bbb75b4c 100644 --- a/server/gpu/modal_deployments/reflector_diarizer.py +++ b/server/gpu/modal_deployments/reflector_diarizer.py @@ -6,7 +6,7 @@ Reflector GPU backend - diarizer import os import modal.gpu -from modal import Image, Secret, App, asgi_app, method, enter +from modal import App, Image, Secret, asgi_app, enter, method from pydantic import BaseModel PYANNOTE_MODEL_NAME: str = "pyannote/speaker-diarization-3.1" @@ -64,9 +64,8 @@ diarizer_image = ( ) ) - @app.cls( - gpu=modal.gpu.A100(memory=40), + gpu=modal.gpu.A100(size="40GB"), timeout=60 * 30, container_idle_timeout=60, allow_concurrent_inputs=1,