Keep production startup independent of benchmarks

This commit is contained in:
Codex
2026-08-15 11:13:29 -07:00
parent 174ba875dd
commit 74a5a252c8
3 changed files with 9 additions and 1 deletions
@@ -7,6 +7,7 @@ from typing import Any
import numpy as np
import torch
import cv2
from .core import normalize_bbox, normalize_keypoints
@@ -45,6 +46,7 @@ class Models:
return digest.hexdigest()
def load(self) -> None:
cv2.setNumThreads(int(os.getenv("CV2_NUM_THREADS", "2")))
torch.backends.cudnn.benchmark = True
torch.backends.cuda.matmul.allow_tf32 = True
torch.backends.cudnn.allow_tf32 = True