update positional_embedding.py

#28

Relates to bf16 for query_states and key_states issue.
Apply fix in https://huggingface.co/microsoft/Phi-3-small-8k-instruct/commit/f196467b67c13127747a03c142e09aa6841447b8 also for this model

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment