Fix LoraEmbedding casting outputs to the token ID dtype#67

Robertboy18 wants to merge 2 commits into

aws-neuron:mainaws-neuron/neuronx-distributed:mainfrom

Robertboy18:fix-lora-embedding-output-dtypeRobertboy18/neuronx-distributed:fix-lora-embedding-output-dtypeCopy head branch name to clipboard