Fixed the test.py to be compatible with older transformers versions e6f6d64 verified ddyuudd commited on 3 days ago