Hugging Face's logo

sanchit-gandhi
/
flax-wav2vec2-2-bart-large

Model card Files Community
sanchit-gandhi HF staff commited on
Commit
83960ec
•
1 Parent(s): 572827c

up

Browse files
Files changed (1) hide show
  1. 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, FlaxSpeechEncoderDecoderModel
 
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"