Downloads · 30 days
4
57% of all-time downloads
crumb/jit-traced-stable-diffusion-unet
jit-traced-stable-diffusion-unet is a text-to-image model from crumb. Use it when you need an image from a text prompt.
Downloads · 30 days
4
57% of all-time downloads
All-time downloads
7
Public
Repo size
1.7 GB
Likes
0
Public
Click a slice to open those files.
.pt1.7 GB · 100%
From the Hugging Face model README
torch.jit.load("unet.pt")
noise_pred = unet(latent_model_input, torch.tensor(t, dtype=torch.float32), text_embeddings) # no ['sample']