new structure for transcription engine

- found problems with mps
- integrated jinja2 templates for rendering (.tex?)
- raw ui & bad structure
This commit is contained in:
2026-02-24 02:58:28 +03:00
parent 9e67b36842
commit 3d48f473b0
24 changed files with 584 additions and 403 deletions
+23
View File
@@ -0,0 +1,23 @@
from dataclasses import dataclass
import torch
@dataclass
class Configuration:
# add new models
device: str = "cuda"
modelName: str = "openai/whisper-large-v2"
chunkSize: int = 30
batchSize: int = 16
dataType: str = "torch.float16"
_dtype_map = {
"torch.float16": torch.float16,
"torch.float32": torch.float32,
"torch.bfloat16": torch.bfloat16,
}
dType: torch.dtype = None
def __post_init__(self):
self.dType = self._dtype_map[self.dataType]