SumNeuron:

It is the same errors. That isn’t the issue.


Using CUDA_LAUNCH_BLOCKING=1 isn’t a fix, but should point to the failing operation in the stacktrace.
Based on your latest update, chck the embedding usage, i.e. its setup and inputs.