Spaces:
Running on Zero
Running on Zero
Vicente Alvarez commited on
Commit ·
435d159
1
Parent(s): c843a44
Preload audio components (audio_encoder, audio_decoder, vocoder)
Browse files
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 |
|