"vscode:/vscode.git/clone" did not exist on "1f8bc2766026951df4b18485bdb6490fe0444b98"
  • Akshay Babbar's avatar
    Fix #12116: preserve boolean dtype for attention masks in ChromaPipeline (#12263) · 0a151115
    Akshay Babbar authored
    
    
    * fix: preserve boolean dtype for attention masks in ChromaPipeline
    
    - Convert attention masks to bool and prevent dtype corruption
    - Fix both positive and negative mask handling in _get_t5_prompt_embeds
    - Remove float conversion in _prepare_attention_mask method
    
    Fixes #12116
    
    * test: add ChromaPipeline attention mask dtype tests
    
    * test: add slow ChromaPipeline attention mask tests
    
    * chore: removed comments
    
    * refactor: removing redundant type conversion
    
    * Remove dedicated dtype tests as per  feedback
    
    ---------
    Co-authored-by: default avatarDhruv Nair <dhruv.nair@gmail.com>
    0a151115
pipeline_chroma.py 44.9 KB