tensorflow has no attribute 'sort' gpt-2 code example

Example: the model tensorflow has no attribute sort

sed -ie s/sort\(/contrib\.framework\.sort\(/g src/sample.py

# For GPT-2 specifically