Replies: 1 comment
-
GPT NEO has its own function. The setup would be like this for Q/A: from quickai import gpt_neo
prompt = """
The bus is a square, red, and has no roof.
What color is the bus?
"""
generated_text = gpt_neo(prompt, "125M")
print(generated_text) It think it should also be able to do French better cause it's a better model. Also, if you want better results, you can change 125M to the larger parameter models. Options are 350M Remember that as you increase the model size, the processing time goes up, but so does the accuracy. The best choice depends on your task. If you want even more parameters, try the gpt_j function(if your system can run this resources wise, you will need nvidia GPU) The code is the same, but there is no model parameter as GPTJ is only 6B. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hi,
I've explore a little set of model, some work out of the box and other complains that they cannot be loaded.
As this model is set in the quickai code, i suppose this is not expected.
Thanks for your help,
BR,
Laurent
Beta Was this translation helpful? Give feedback.
All reactions