xtuner icon indicating copy to clipboard operation
xtuner copied to clipboard

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

Open xiaohangguo opened this issue 11 months ago • 4 comments


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


· 暂且放在了tools里面。

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

xiaohangguo avatar Mar 14 '24 07:03 xiaohangguo

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

pppppM avatar Mar 14 '24 07:03 pppppM

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

pppppM avatar Mar 14 '24 07:03 pppppM

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

ok,我写了一下help

xiaohangguo avatar Mar 14 '24 07:03 xiaohangguo

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

xiaohangguo avatar Mar 14 '24 09:03 xiaohangguo