text_embeddings = torch.cat([uncond_embeddings, text_embeddings])