qkv_out can be a single tensor or a list. Handling these cases separe…#1850

jeffra merged 2 commits intomasterdeepspeedai/DeepSpeed:masterfrom

samyamr/small-inference-bugdeepspeedai/DeepSpeed:samyamr/small-inference-bugCopy head branch name to clipboard