• Pedro Cuenca's avatar
    Continuation of #942: additional float64 failure (#996) · 1d04e1b4
    Pedro Cuenca authored
    * Add failing test for #940.
    
    * Do not use torch.float64 in mps.
    
    * style
    
    * Temporarily skip add_noise for IPNDMScheduler.
    
    Until #990 is addressed.
    
    * Fix additional float64 error in mps.
    
    * Improve add_noise test
    
    * Slight edit – I think it's clearer this way.
    1d04e1b4
scheduling_lms_discrete.py 12.2 KB