pe —
data type of the generated embeddings   Returns 
torch.FloatTensor
  
Embedding vectors with shape (len(timesteps), embedding_dim)
  See https://github.com/google-research/vdm/blob/dc27b98a554f65cdc654b800da5aa1846545d41b/model_vdm.py#L298  unfuse_qkv_projections  < source > ( unet: bool = True vae: bool = True  )   Parameters   unet (bool, defaults to True) — To apply fusion on the UNet.   vae (bool, defaults to True) — To apply fusion on