Update utils.py

utils.py

@@ -275,7 +275,9 @@ def gen_image_as_per_prompt(prompt, style, seed, custom_loss=None):
         # replacement_token_embedding = text_encoder.get_input_embeddings()(torch.tensor(2368, device=torch_device))
 
         # Insert this into the token embeddings (
-        token_embeddings[0, torch.where(input_ids[0] == 6829)] = replacement_token_embedding.to(torch_device)
+        indices = torch.where(input_ids[0] == 6829)[0]  # Extract indices where the condition is true
+        if indices.numel() > 0:  # Check if any indices are found
+             token_embeddings[0, indices] = replacement_token_embedding.to(torch_device)
 
     # get pos embed
     pos_emb_layer = text_encoder.text_model.embeddings.position_embedding