search

thadunge2 / AIDungeon

Other
51 stars 32 forks source link

corrected requirements for pytorch model #54

Closed ShnitzelKiller closed 4 years ago

ShnitzelKiller commented 4 years ago

it was missing transformers and still depended on TF

thadunge2 commented 4 years ago

You need TF2.0 to convert TF models into Torch. Also, transformers isn't a requirement because you have to do a custom install of it -- check the bottom of play.py in the branch. All you have to do is run play.py and it will install everything.

ShnitzelKiller commented 4 years ago

FYI if transformers isn't installed the script won't get around to prepare_aidungeon_2().