• Sayak Paul's avatar
    [Tests] Speed up example tests (#6319) · f4b0b26f
    Sayak Paul authored
    * remove validation args from textual onverson tests
    
    * reduce number of train steps in textual inversion tests
    
    * fix: directories.
    
    * debig
    
    * fix: directories.
    
    * remove validation tests from textual onversion
    
    * try reducing the time of test_text_to_image_checkpointing_use_ema
    
    * fix: directories
    
    * speed up test_text_to_image_checkpointing
    
    * speed up test_text_to_image_checkpointing_checkpoints_total_limit_removes_multiple_checkpoints
    
    * fix
    
    * speed up test_instruct_pix2pix_checkpointing_checkpoints_total_limit_removes_multiple_checkpoints
    
    * set checkpoints_total_limit to 2.
    
    * test_text_to_image_lora_checkpointing_checkpoints_total_limit_removes_multiple_checkpoints speed up
    
    * speed up test_unconditional_checkpointing_checkpoints_total_limit_removes_multiple_checkpoints
    
    * debug
    
    * fix: directories.
    
    * speed up test_instruct_pix2pix_checkpointing_checkpoints_total_limit
    
    * speed up: test_controlnet_checkpointing_checkpoints_total_limit_removes_multiple_checkpoints
    
    * speed up test_controlnet_sdxl
    
    * speed up dreambooth tests
    
    * speed up test_dreambooth_lora_checkpointing_checkpoints_total_limit_removes_multiple_checkpoints
    
    * speed up test_custom_diffusion_checkpointing_checkpoints_total_limit_removes_multiple_checkpoints
    
    * speed up test_text_to_image_lora_sdxl_text_encoder_checkpointing_checkpoints_total_limit
    
    * speed up # checkpoint-2 should have been deleted
    
    * speed up examples/text_to_image/test_text_to_image.py::TextToImage::test_text_to_image_checkpointing_checkpoints_total_limit
    
    * additional speed ups
    
    * style
    f4b0b26f
test_custom_diffusion.py 4.67 KB