• Joao Gante's avatar
    Add TFSpeech2Text (#15113) · 8406fa6d
    Joao Gante authored
    * Add wrapper classes
    
    * convert inner layers to tf
    
    * Add TF Encoder and Decoder layers
    
    * TFSpeech2Text models
    
    * Loadable model
    
    * TF model with same outputs as PT model
    
    * test skeleton
    
    * correct tests and run the fixup
    
    * correct attention expansion
    
    * TFSpeech2Text pask_key_values with TF format
    8406fa6d
dummy_tf_objects.py 45.8 KB