Upload DiCoWForConditionalGeneration
Browse files- config.json +1 -1
- config.py +1 -1
config.json
CHANGED
|
@@ -56,7 +56,7 @@
|
|
| 56 |
"max_source_positions": 1500,
|
| 57 |
"max_target_positions": 448,
|
| 58 |
"median_filter_width": 7,
|
| 59 |
-
"model_type": "
|
| 60 |
"mt_num_speakers": 1,
|
| 61 |
"n_soft_prompts": 16,
|
| 62 |
"non_target_fddt_value": 0.5,
|
|
|
|
| 56 |
"max_source_positions": 1500,
|
| 57 |
"max_target_positions": 448,
|
| 58 |
"median_filter_width": 7,
|
| 59 |
+
"model_type": "DiCoW",
|
| 60 |
"mt_num_speakers": 1,
|
| 61 |
"n_soft_prompts": 16,
|
| 62 |
"non_target_fddt_value": 0.5,
|
config.py
CHANGED
|
@@ -26,7 +26,7 @@ class Seq2SeqModelOutputLogit(Seq2SeqModelOutput):
|
|
| 26 |
class DiCoWConfig(WhisperConfig):
|
| 27 |
"""This is a modified version of the `WhisperEncoder` model from the `transformers` library.
|
| 28 |
The model has been modified to support CTC loss computation in the forward pass."""
|
| 29 |
-
|
| 30 |
def __init__(
|
| 31 |
self,
|
| 32 |
ctc_loss_reduction: str = "mean",
|
|
|
|
| 26 |
class DiCoWConfig(WhisperConfig):
|
| 27 |
"""This is a modified version of the `WhisperEncoder` model from the `transformers` library.
|
| 28 |
The model has been modified to support CTC loss computation in the forward pass."""
|
| 29 |
+
model_type = "DiCoW"
|
| 30 |
def __init__(
|
| 31 |
self,
|
| 32 |
ctc_loss_reduction: str = "mean",
|