• Sayak Paul's avatar
    [Tests] Speed up some fast pipeline tests (#7477) · fac76169
    Sayak Paul authored
    * speed up test_vae_slicing in animatediff
    
    * speed up test_karras_schedulers_shape for attend and excite.
    
    * style.
    
    * get the static slices out.
    
    * specify torch print options.
    
    * modify
    
    * test run with controlnet
    
    * specify kwarg
    
    * fix: things
    
    * not None
    
    * flatten
    
    * controlnet img2img
    
    * complete controlet sd
    
    * finish more
    
    * finish more
    
    * finish more
    
    * finish more
    
    * finish the final batch
    
    * add cpu check for expected_pipe_slice.
    
    * finish the rest
    
    * remove print
    
    * style
    
    * fix ssd1b controlnet test
    
    * checking ssd1b
    
    * disable the test.
    
    * make the test_ip_adapter_single controlnet test more robust
    
    * fix: simple inpaint
    
    * multi
    
    * disable panorama
    
    * enable again
    
    * panorama is shaky so leave it for now
    
    * remove print
    
    * raise tolerance.
    fac76169
test_stable_diffusion.py 57 KB