svlqd
3d48f473b0
- found problems with mps - integrated jinja2 templates for rendering (.tex?) - raw ui & bad structure
30 lines
572 B
Python
30 lines
572 B
Python
import torch
|
|
from abc import ABC, abstractmethod
|
|
|
|
class BaseEngine(ABC):
|
|
def __init__(
|
|
self,
|
|
modelName: str,
|
|
language: str,
|
|
dType: torch.dtype,
|
|
device: str
|
|
):
|
|
self.modelName = modelName
|
|
self.device = device
|
|
self.language = language
|
|
self.dType = dType
|
|
|
|
@abstractmethod
|
|
def loadModel(self) -> None:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def unloadModel(self) -> None:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def transcribeBatch(
|
|
self,
|
|
batch
|
|
) -> str:
|
|
pass |