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

Downloads last month: -; Downloads are not tracked for this model. How to track

Inference Providers NEW

This model is not currently available via any of the supported Inference Providers.

The model cannot be deployed to the HF Inference API: The model authors have turned it off explicitly.