DeepSpeed

e0dd488f - fix layer_past; this caused issues when running inference on several queries