-
Notifications
You must be signed in to change notification settings - Fork 1.7k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
生成的速度太慢了,能否加一个生成的batch_size大于1的功能 #45
Comments
generate函数的确有优化的空间,没有用到past,利用到past,速度还快很多 |
merge了fengzuo97的PR,试试看呢? |
我想问问你们的数据集从哪里来的呢 能上传一个数据集的样本吗 |
中文数据集在项目README文档中有 |
要是有教程把 GPT2-ML 的1.5 中文模型 转成本项目的Pytorch格式就好了 |
我试过,里面缺个东西,目前转不了,不信你可以试试下面的代码: import argparse import torch from transformers import CONFIG_NAME, WEIGHTS_NAME, GPT2Config, GPT2Model, load_tf_weights_in_gpt2 logging.basicConfig(level=logging.INFO) def convert_gpt2_checkpoint_to_pytorch(gpt2_checkpoint_path, gpt2_config_file, pytorch_dump_folder_path):
if name == "main": |
如题!谢谢!
The text was updated successfully, but these errors were encountered: