remove re-defination of FlaxWav2Vec2ForCTCModule (#15965)

This commit is contained in:
Suraj Patil 2022-03-07 14:58:44 +01:00 committed by GitHub
parent ef9c3ca348
commit 60b81dfa6f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1206,10 +1206,6 @@ overwrite_call_docstring(
append_replace_return_docstrings(FlaxWav2Vec2ForCTC, output_type=FlaxCausalLMOutput, config_class=Wav2Vec2Config)
class FlaxWav2Vec2ForCTCModule(nn.Module):
config: Wav2Vec2Config
class FlaxWav2Vec2ForPreTrainingModule(nn.Module):
config: Wav2Vec2Config
dtype: jnp.dtype = jnp.float32
@ -1409,7 +1405,3 @@ overwrite_call_docstring(
append_replace_return_docstrings(
FlaxWav2Vec2ForPreTraining, output_type=FlaxWav2Vec2ForPreTrainingOutput, config_class=Wav2Vec2Config
)
class FlaxWav2Vec2ForCTCModule(nn.Module):
config: Wav2Vec2Config