Commit
•
83960ec
1
Parent(s):
572827c
up
Browse files- create_model.py +2 -1
create_model.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
import jax
|
2 |
import jax.numpy as jnp
|
3 |
-
from transformers import AutoFeatureExtractor, AutoTokenizer
|
|
|
4 |
|
5 |
encoder_id = "facebook/wav2vec2-large-lv60"
|
6 |
decoder_id = "facebook/bart-large"
|
|
|
1 |
import jax
|
2 |
import jax.numpy as jnp
|
3 |
+
from transformers import AutoFeatureExtractor, AutoTokenizer
|
4 |
+
from models.modeling_flax_speech_encoder_decoder import FlaxSpeechEncoderDecoderModel
|
5 |
|
6 |
encoder_id = "facebook/wav2vec2-large-lv60"
|
7 |
decoder_id = "facebook/bart-large"
|