Update custom model files, README, and requirements
Browse files- asr_modeling.py +0 -2
asr_modeling.py
CHANGED
|
@@ -38,8 +38,6 @@ class ASRModel(PreTrainedModel, GenerationMixin):
|
|
| 38 |
@classmethod
|
| 39 |
def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
|
| 40 |
"""Load model from pretrained, handling device placement correctly."""
|
| 41 |
-
from pathlib import Path
|
| 42 |
-
|
| 43 |
from safetensors.torch import load_file
|
| 44 |
from transformers.utils.hub import cached_file
|
| 45 |
|
|
|
|
| 38 |
@classmethod
|
| 39 |
def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
|
| 40 |
"""Load model from pretrained, handling device placement correctly."""
|
|
|
|
|
|
|
| 41 |
from safetensors.torch import load_file
|
| 42 |
from transformers.utils.hub import cached_file
|
| 43 |
|