• Sayak Paul's avatar
    [I2VGenXL] `attention_head_dim` in the UNet (#6872) · 491a933a
    Sayak Paul authored
    * attention_head_dim
    
    * debug
    
    * print more info
    
    * correct num_attention_heads behaviour
    
    * down_block_num_attention_heads -> num_attention_heads.
    
    * correct the image link in doc.
    
    * add: deprecation for num_attention_head
    
    * fix: test argument to use attention_head_dim
    
    * more fixes.
    
    * quality
    
    * address comments.
    
    * remove depcrecation.
    491a933a
attention.py 27.5 KB