How to stop model generating
jingrongchen opened this issue · comments
John Chen commented
I have finetuned my model using my own dataset, but when i run the model, it generates a correct answer but the model keep on generating the prompt answer pair, is there any way to tell model to stop the genration after certain character
Sri Aravind Sairaman commented
Hi, I am trying to finetune the model on my own dataset as well. But even with GPU memory of 24GB, I had issues loading the model weights into it. What would be the suggestion for the compute hardware for finetuning GPTJ 6B.
I am training with input data with an avg of 660 words.