This commit is contained in:
AI-Casanova
2024-01-27 12:15:38 -06:00
committed by Vladimir Mandic
parent 5c78268e28
commit a614170f50
@@ -110,9 +110,7 @@ def convert_embedding(tensor, text_encoder, text_encoder_2):
indices *= 0 # Use SDXL padding vector 0
vectors.append(indices)
vectors = torch.stack(vectors)
print(vectors)
output = text_encoder_2.get_input_embeddings().weight.data[vectors]
print(output.shape)
return output