add type Callable
Browse files
model.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
from transformers.models.whisper.configuration_whisper import WhisperConfig
|
2 |
|
3 |
-
from typing import List, Optional, Dict, Any
|
4 |
|
5 |
"""Custom config to support modification of the Whisper encoder."""
|
6 |
|
|
|
1 |
from transformers.models.whisper.configuration_whisper import WhisperConfig
|
2 |
|
3 |
+
from typing import List, Optional, Dict, Any, Callable
|
4 |
|
5 |
"""Custom config to support modification of the Whisper encoder."""
|
6 |
|