Skip to content

GitLab

Unverified Commit 60b81dfa authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

remove re-defination of FlaxWav2Vec2ForCTCModule (#15965)

parent ef9c3ca3
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 8 deletions
src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py
View file @ 60b81dfa
...@@ -1206,10 +1206,6 @@ overwrite_call_docstring( ...@@ -1206,10 +1206,6 @@ overwrite_call_docstring(
append_replace_return_docstrings(FlaxWav2Vec2ForCTC, output_type=FlaxCausalLMOutput, config_class=Wav2Vec2Config) append_replace_return_docstrings(FlaxWav2Vec2ForCTC, output_type=FlaxCausalLMOutput, config_class=Wav2Vec2Config)
class FlaxWav2Vec2ForCTCModule(nn.Module):
config: Wav2Vec2Config
class FlaxWav2Vec2ForPreTrainingModule(nn.Module): class FlaxWav2Vec2ForPreTrainingModule(nn.Module):
config: Wav2Vec2Config config: Wav2Vec2Config
dtype: jnp.dtype = jnp.float32 dtype: jnp.dtype = jnp.float32
...@@ -1409,7 +1405,3 @@ overwrite_call_docstring( ...@@ -1409,7 +1405,3 @@ overwrite_call_docstring(
append_replace_return_docstrings( append_replace_return_docstrings(
FlaxWav2Vec2ForPreTraining, output_type=FlaxWav2Vec2ForPreTrainingOutput, config_class=Wav2Vec2Config FlaxWav2Vec2ForPreTraining, output_type=FlaxWav2Vec2ForPreTrainingOutput, config_class=Wav2Vec2Config
) )
class FlaxWav2Vec2ForCTCModule(nn.Module):
config: Wav2Vec2Config
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment