StableLM icon indicating copy to clipboard operation
StableLM copied to clipboard

Dataset used to pre-train

Open agademic opened this issue 2 years ago • 5 comments

Hi there! First of all, thank you for the amazing work! The readme says the models were trained on "the new dataset based on The Pile" which is 3x the size of The Pile. Can you give more insights on the dataset and its content?

Thank you!

agademic avatar Apr 19 '23 17:04 agademic

The readme indicates they are planning to release a technical report soon. I suspect the details will be in the tech report. I also hope they continue the training on the 3b and 7b model all the way up to 1.5T tokens!

THANK YOU STABILITY AI Your contributions to the Open Source community are very much appreciated!

gururise avatar Apr 19 '23 17:04 gururise

Can we expect that this forthcoming dataset declaration will include those inputs that imbue this model with politically correct output (even with a neutral SYSTEM prompt) ?

fche avatar Apr 19 '23 18:04 fche

Can we expect that this forthcoming dataset declaration will include those inputs that imbue this model with politically correct output (even with a neutral SYSTEM prompt) ?

Only the "Tuned" model has a SYSTEM prompt and that model's finetuning datasets are where that is coming from. They're the same finetuning data used for llama finetunes like Alpaca and GPT4All, which are outputs of ChatGPT. So ultimately they come from ChatGPT.

The "Base" model does not have a SYSTEM prompt and does not use those datasets or any like them.

MarkSchmidty avatar Apr 20 '23 07:04 MarkSchmidty

It would really be interesting and important to share some basic information about the training data already and in the repo though, especially the kind and size of training data for each language. Judging from a few first trials, the amount of German training data is probably very small and thus results in German are quite poor.

johann-petrak avatar Apr 21 '23 09:04 johann-petrak

More information will be published soon and is very likely to answer your questions when it's available.

mcmonkey4eva avatar Apr 24 '23 15:04 mcmonkey4eva