• Sanchit Gandhi's avatar
    Add FlaxBartForCausalLM (#15995) · b256f351
    Sanchit Gandhi authored
    * add causal lm
    
    * add CausalLM tests
    
    * Add FlaxBartForCausalLM
    
    * Add EncoderDecoder model tests
    
    * change docstring
    
    * make repo-consistency
    
    * suggested changes
    
    * remove jax ops
    
    * correction
    
    * rename pre-trained decoder model
    b256f351
check_repo.py 29.4 KB