dagloop5 commited on
Commit
5ff59d3
·
verified ·
1 Parent(s): ed341d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -385,7 +385,7 @@ _orig_audio_decoder_factory1 = ledger1.audio_decoder
385
  _orig_vocoder_factory1 = ledger1.vocoder
386
  _orig_spatial_upsampler_factory1 = ledger1.spatial_upsampler
387
  _orig_text_encoder_factory1 = ledger1.text_encoder
388
- _orig_embeddings_processor_factory1 = ledger1.embeddings_processor
389
 
390
  _transformer1 = _orig_transformer_factory1()
391
  _video_encoder1 = _orig_video_encoder_factory1()
@@ -394,7 +394,7 @@ _audio_decoder1 = _orig_audio_decoder_factory1()
394
  _vocoder1 = _orig_vocoder_factory1()
395
  _spatial_upsampler1 = _orig_spatial_upsampler_factory1()
396
  _text_encoder1 = _orig_text_encoder_factory1()
397
- _embeddings_processor1 = _orig_embeddings_processor_factory1()
398
 
399
  ledger1.transformer = lambda: _transformer1
400
  ledger1.video_encoder = lambda: _video_encoder1
@@ -403,7 +403,7 @@ ledger1.audio_decoder = lambda: _audio_decoder1
403
  ledger1.vocoder = lambda: _vocoder1
404
  ledger1.spatial_upsampler = lambda: _spatial_upsampler1
405
  ledger1.text_encoder = lambda: _text_encoder1
406
- ledger1.embeddings_processor = lambda: _embeddings_processor1
407
 
408
  print(" Stage 1 models preloaded")
409
 
@@ -418,7 +418,7 @@ _orig_audio_decoder_factory2 = ledger2.audio_decoder
418
  _orig_vocoder_factory2 = ledger2.vocoder
419
  _orig_spatial_upsampler_factory2 = ledger2.spatial_upsampler
420
  _orig_text_encoder_factory2 = ledger2.text_encoder
421
- _orig_embeddings_processor_factory2 = ledger2.embeddings_processor
422
 
423
  _transformer2 = _orig_transformer_factory2()
424
  _video_encoder2 = _orig_video_encoder_factory2()
@@ -427,7 +427,7 @@ _audio_decoder2 = _orig_audio_decoder_factory2()
427
  _vocoder2 = _orig_vocoder_factory2()
428
  _spatial_upsampler2 = _orig_spatial_upsampler_factory2()
429
  _text_encoder2 = _orig_text_encoder_factory2()
430
- _embeddings_processor2 = _orig_embeddings_processor_factory2()
431
 
432
  ledger2.transformer = lambda: _transformer2
433
  ledger2.video_encoder = lambda: _video_encoder2
@@ -436,7 +436,7 @@ ledger2.audio_decoder = lambda: _audio_decoder2
436
  ledger2.vocoder = lambda: _vocoder2
437
  ledger2.spatial_upsampler = lambda: _spatial_upsampler2
438
  ledger2.text_encoder = lambda: _text_encoder2
439
- ledger2.embeddings_processor = lambda: _embeddings_processor2
440
 
441
  print(" Stage 2 models preloaded")
442
 
 
385
  _orig_vocoder_factory1 = ledger1.vocoder
386
  _orig_spatial_upsampler_factory1 = ledger1.spatial_upsampler
387
  _orig_text_encoder_factory1 = ledger1.text_encoder
388
+ _orig_gemma_embeddings_factory1 = ledger1.gemma_embeddings_processor
389
 
390
  _transformer1 = _orig_transformer_factory1()
391
  _video_encoder1 = _orig_video_encoder_factory1()
 
394
  _vocoder1 = _orig_vocoder_factory1()
395
  _spatial_upsampler1 = _orig_spatial_upsampler_factory1()
396
  _text_encoder1 = _orig_text_encoder_factory1()
397
+ _embeddings_processor1 = _orig_gemma_embeddings_factory1()
398
 
399
  ledger1.transformer = lambda: _transformer1
400
  ledger1.video_encoder = lambda: _video_encoder1
 
403
  ledger1.vocoder = lambda: _vocoder1
404
  ledger1.spatial_upsampler = lambda: _spatial_upsampler1
405
  ledger1.text_encoder = lambda: _text_encoder1
406
+ ledger1.gemma_embeddings_processor = lambda: _embeddings_processor1
407
 
408
  print(" Stage 1 models preloaded")
409
 
 
418
  _orig_vocoder_factory2 = ledger2.vocoder
419
  _orig_spatial_upsampler_factory2 = ledger2.spatial_upsampler
420
  _orig_text_encoder_factory2 = ledger2.text_encoder
421
+ _orig_gemma_embeddings_factory2 = ledger2.gemma_embeddings_processor
422
 
423
  _transformer2 = _orig_transformer_factory2()
424
  _video_encoder2 = _orig_video_encoder_factory2()
 
427
  _vocoder2 = _orig_vocoder_factory2()
428
  _spatial_upsampler2 = _orig_spatial_upsampler_factory2()
429
  _text_encoder2 = _orig_text_encoder_factory2()
430
+ _embeddings_processor2 = _orig_gemma_embeddings_factory2()
431
 
432
  ledger2.transformer = lambda: _transformer2
433
  ledger2.video_encoder = lambda: _video_encoder2
 
436
  ledger2.vocoder = lambda: _vocoder2
437
  ledger2.spatial_upsampler = lambda: _spatial_upsampler2
438
  ledger2.text_encoder = lambda: _text_encoder2
439
+ ledger2.gemma_embeddings_processor = lambda: _embeddings_processor2
440
 
441
  print(" Stage 2 models preloaded")
442