模型:
merve/chatgpt-prompt-generator-v12
此模型是 ChatGPT 提示生成器数据集上 BART-large 的经过微调的版本。它在评估集上达到以下结果:
您可以使用此模型生成 ChatGPT 的个人资料。只需输入如下的个人信息:
from transformers import BartForConditionalGeneration, BartTokenizer example_english_phrase = "photographer" batch = tokenizer(example_english_phrase, return_tensors="pt") generated_ids = model.generate(batch["input_ids"], max_new_tokens=150) output = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
训练时使用了以下超参数:
Train Loss | Validation Loss | Epoch |
---|---|---|
5.3808 | 3.3133 | 0 |
3.2642 | 3.0104 | 1 |
2.8886 | 2.8600 | 2 |
2.6594 | 2.7949 | 3 |
2.4800 | 2.7320 | 4 |