Fix for #91 for newer versions of tensorflow#94

bcazur wants to merge 2 commits intokpe:masterkpe/bert-for-tf2:masterfrom

bcazur:masterbcazur/bert-for-tf2:masterCopy head branch name to clipboard