Minor improvements & docs added

This commit is contained in:
2025-11-04 15:57:52 +03:00
parent 633ac7e081
commit 7f92a27b49
7 changed files with 58 additions and 62 deletions
+3 -2
View File
@@ -17,6 +17,7 @@ class DeviceConfiguration:
- "openai/whisper-medium"
- "openai/whisper-large"
- "openai/whisper-large-v2"
- "openai/whisper-large-v3-turbo"
batch_size (int): Chunks in one batch. Selected for VRAM.
@@ -24,8 +25,8 @@ class DeviceConfiguration:
data_type (str): custom data type of model. Variants:
- torch.float16 - for GPUs
- torch.float32 - for CPU / weak GPU
- torch.bfloat16 - for GPUs which has BF16 support
- torch.float32 - for CPU
- torch.bfloat16 - for GPUs which has BF16 support (usually RTX 40XX+)
"""
device: str = "cuda"