Message Boards Message Boards

GPT-2 language model size issues, importing neural networks

Posted 4 years ago

I have not done much with neural networks in the past, but the Wolfram Language made it easy for me to jump right in. This page: made setting up a language model for text completion as simple as copy-paste! I've been having fun messing around with the model.

But I have an issue now. As you can see from the page I posted, the built in model lets you choose a "size" for the model: one can currently choose between 117M and 335M. As you can see from this page: the researchers who made the model released three sizes of their model, 117M, 335M, and 774M. For their own reasons they have not released the full model yet. However, another group has apparently replicated their full 1.5B model anyway: They have links in their first paragraph to the model, apparently.

I've messed around with the first three sizes of models on a website that implemented them, and in my opinion there seems to be quite a difference in how it performs when the dataset gets larger. I would love to try out the 1.5B model, which was not on the website I tried, and is not , apparently, built into the Wolfram Language at this time. Even having access to the 774M model in the Wolfram Lanugage would be nice.

Does anyone know how to go about importing the larger models from those websites into the Wolfram Lanuguage and then running them? I did try downloading some files from the website with the larger model and using the "Import" function, but didn't have much luck with that as it didn't seem to recognize the file format. Again, I feel like I'm pretty good with WL, (wolfram language), but I'm not too proficient with the details of neural networks, I'm just pluggin stuff in as is suggested on that Wolfram page about GPT-2. Alternatively if anyone happened to know anything about if Wolfram Research plans to implement the larger model(s) any time soon that might be interesting to know as well.


POSTED BY: Eric Parfitt
7 Replies
Posted 1 year ago

Hi, I noticed the GPT2 1.5B model is still not part of the repository.

Are there plans to add it? I assume its absence would be noted, given the wide interest in GPT models.

Posted 4 years ago

Yes, these models have been converted. I will try to publish them as soon as possible.

Thanks for inquiring again (and being impatient in this case).

POSTED BY: Test Account

I haven't seen the larger models posted yet. any chance those can get up?

Hi, apologies if I'm being impatient. I hear you that you guys have other priorities for neural networks and all the other things Wolfram Research is doing. I'm just wondering if there's any update thanks :-/ I'm working on a project that uses GPT-2 and getting some interesting results. I'll post my code when I'm finished, but ofc I just can't wait to plug in a more powerful language generator into the code X-D

If anyone had tips for importing the full GPT-2 dataset in another way either, that would also be much appreciated. Otherwise I could look into finding a way to link it myself I guess but I'm laazy lol. That's what Wolfram Language is great for, just plugging stuff in that just works X-D Thanks you guys.


POSTED BY: Eric Parfitt
Posted 4 years ago


Thank you for your suggestions.

When we first published the models, only 117M and 334M was released and as a result, I imported just those two. I will make sure to add the remainder of the models, and put it as a suggestion.

Thanks, Tuseeta

POSTED BY: Test Account

Great to hear! Thanks! Nice that you guys are keeping on top of things, the new model hasn't been out that long and I'm sure there's lots of other stuff to add too, so truly thanks :-)

POSTED BY: Eric Parfitt
Posted 4 years ago

Yes, while I was aware of the newer/bigger model releases, we were focusing on other models/improving or introducing new functionalities that would allow for importing more models. However, I will make sure that the bigger models are also added in to-do list or imported.

POSTED BY: Test Account
Reply to this discussion
Community posts can be styled and formatted using the Markdown syntax.
Reply Preview
or Discard

Group Abstract Group Abstract