Josephanthraper commited on
Commit
8ddb8f4
·
verified ·
1 Parent(s): a9ca0d6

Update whisper_jax/__init__.py

Browse files
Files changed (1) hide show
  1. whisper_jax/__init__.py +3 -3
whisper_jax/__init__.py CHANGED
@@ -15,7 +15,7 @@
15
 
16
  __version__ = "0.0.1"
17
 
18
- from .modeling_flax_whisper import FlaxWhisperForConditionalGeneration
19
- from .partitioner import PjitPartitioner
20
  from .pipeline import FlaxWhisperPipline
21
- from .train_state import InferenceState
 
15
 
16
  __version__ = "0.0.1"
17
 
18
+ #from .modeling_flax_whisper import FlaxWhisperForConditionalGeneration
19
+ #from .partitioner import PjitPartitioner
20
  from .pipeline import FlaxWhisperPipline
21
+ #from .train_state import InferenceState