• Yuxuan.Zhang's avatar
    CogView3Plus DiT (#9570) · 8d81564b
    Yuxuan.Zhang authored
    * merge 9588
    
    * max_shard_size="5GB" for colab running
    
    * conversion script updates; modeling test; refactor transformer
    
    * make fix-copies
    
    * Update convert_cogview3_to_diffusers.py
    
    * initial pipeline draft
    
    * make style
    
    * fight bugs 🐛
    
    🪳
    
    * add example
    
    * add tests; refactor
    
    * make style
    
    * make fix-copies
    
    * add co-author
    
    YiYi Xu <yixu310@gmail.com>
    
    * remove files
    
    * add docs
    
    * add co-author
    Co-Authored-By: default avatarYiYi Xu <yixu310@gmail.com>
    
    * fight docs
    
    * address reviews
    
    * make style
    
    * make model work
    
    * remove qkv fusion
    
    * remove qkv fusion tets
    
    * address review comments
    
    * fix make fix-copies error
    
    * remove None and TODO
    
    * for FP16(draft)
    
    * make style
    
    * remove dynamic cfg
    
    * remove pooled_projection_dim as a parameter
    
    * fix tests
    
    ---------
    Co-authored-by: default avatarAryan <aryan@huggingface.co>
    Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
    8d81564b
__init__.py 27.5 KB