Skip to content
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

plug py moudle : Serial merge pt weights to form pth, transform to hf format #476

Open
wants to merge 26 commits into
base: main
Choose a base branch
from

Conversation

xiaohangguo
Copy link
Contributor


py小脚本,串行合并训练好的pt权重形成pth,pth会输出在ckpt_dir同级目录下,最终调用pth_to_hf转化成hf格式。


· 暂且放在了tools里面。

· 存放项目位置和命名觉得不妥可能要修改一下。

@pppppM
Copy link
Collaborator

pppppM commented Mar 14, 2024

File Changed 里有些问题,有 dataset 相关的改动

@pppppM
Copy link
Collaborator

pppppM commented Mar 14, 2024

最好添加一些注释,展示一下输入 ckpt dir 的格式

@xiaohangguo
Copy link
Contributor Author

最好添加一些注释,展示一下输入 ckpt dir 的格式

ok,我写了一下help

@xiaohangguo
Copy link
Contributor Author

@pppppM 佬,是不是还得在哪加个readme.md?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants