• Jianghai's avatar
    [pipeline] All bert models (#4233) · e7cc62d7
    Jianghai authored
    * bloom policy
    
    * llama pipeline forward and tests
    
    * fix the output and attention_mask
    
    * fix name
    
    * bind argument to policy
    
    * Revert "bloom policy"
    
    This reverts commit 8dee68a0a22568dbeed6d4563372b25e1e825fb0.
    
    This policy should be revert and copied to feature/bloom
    
    * revert the bloom changes
    
    * cancel unneeded inputs
    
    * gpt
    
    * finish llama
    
    * causal lm and sequence classification
    
    * revision
    
    * add pure pipeline test
    
    * finish some bert models
    
    * finish all bert models
    
    * finish bert tests
    
    * fix bugs
    
    * fix bugs
    
    * fix test pipeline
    
    * fix data gen for qa
    
    * update the set pipeline forward
    
    * shared params
    
    * fix bugs
    e7cc62d7
_utils.py 2.42 KB