• Dhruv Nair's avatar
    Test Cleanup Precision issues (#4812) · 189e9f01
    Dhruv Nair authored
    
    
    * proposal for flaky tests
    
    * more precision fixes
    
    * move more tests to use cosine distance
    
    * more test fixes
    
    * clean up
    
    * use default attn
    
    * clean up
    
    * update expected value
    
    * make style
    
    * make style
    
    * Apply suggestions from code review
    
    * Update src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_img2img.py
    
    * make style
    
    * fix failing tests
    
    ---------
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    189e9f01
test_models_unet_2d.py 9.72 KB