• Suraj Patil's avatar
    [img2img, inpainting] fix fp16 inference (#769) · 92d70863
    Suraj Patil authored
    * handle dtype in vae and image2image pipeline
    
    * fix inpaint in fp16
    
    * dtype should be handled in add_noise
    
    * style
    
    * address review comments
    
    * add simple fast tests to check fp16
    
    * fix test name
    
    * put mask in fp16
    92d70863
vae.py 21.2 KB