end_latent = end_latent.to(dtype=torch.float32, device=cpu) a0636cb verified Fabrice-TIERCELIN commited on 11 days ago