Vicente Alvarez commited on
Commit
435d159
·
1 Parent(s): c843a44

Preload audio components (audio_encoder, audio_decoder, vocoder)

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -160,6 +160,9 @@ _video_decoder = ledger.video_decoder()
160
  _spatial_upsampler = ledger.spatial_upsampler()
161
  _text_encoder = ledger.text_encoder()
162
  _embeddings_processor = ledger.gemma_embeddings_processor()
 
 
 
163
 
164
  # Replace ledger methods with lambdas returning preloaded instances
165
  ledger.transformer = lambda: _transformer
@@ -168,6 +171,9 @@ ledger.video_decoder = lambda: _video_decoder
168
  ledger.spatial_upsampler = lambda: _spatial_upsampler
169
  ledger.text_encoder = lambda: _text_encoder
170
  ledger.gemma_embeddings_processor = lambda: _embeddings_processor
 
 
 
171
 
172
  print("All models preloaded!")
173
 
 
160
  _spatial_upsampler = ledger.spatial_upsampler()
161
  _text_encoder = ledger.text_encoder()
162
  _embeddings_processor = ledger.gemma_embeddings_processor()
163
+ _audio_encoder = ledger.audio_encoder()
164
+ _audio_decoder = ledger.audio_decoder()
165
+ _vocoder = ledger.vocoder()
166
 
167
  # Replace ledger methods with lambdas returning preloaded instances
168
  ledger.transformer = lambda: _transformer
 
171
  ledger.spatial_upsampler = lambda: _spatial_upsampler
172
  ledger.text_encoder = lambda: _text_encoder
173
  ledger.gemma_embeddings_processor = lambda: _embeddings_processor
174
+ ledger.audio_encoder = lambda: _audio_encoder
175
+ ledger.audio_decoder = lambda: _audio_decoder
176
+ ledger.vocoder = lambda: _vocoder
177
 
178
  print("All models preloaded!")
179