Views
No views yet
1torch.jit.load("unet.pt") 2noise_pred = unet(latent_model_input, torch.tensor(t, dtype=torch.float32), text_embeddings) # no ['sample']