{"id":19054693,"url":"https://github.com/ssbuild/rwkv_finetuning","last_synced_at":"2025-04-24T03:27:39.467Z","repository":{"id":172952502,"uuid":"648672347","full_name":"ssbuild/rwkv_finetuning","owner":"ssbuild","description":"rwkv finetuning","archived":false,"fork":false,"pushed_at":"2024-04-22T01:31:38.000Z","size":1381,"stargazers_count":36,"open_issues_count":0,"forks_count":1,"subscribers_count":1,"default_branch":"dev","last_synced_at":"2025-04-18T12:17:53.565Z","etag":null,"topics":["lora","p-tuning-v2","qlora","rwkv","rwkv4"],"latest_commit_sha":null,"homepage":"","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"apache-2.0","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/ssbuild.png","metadata":{"files":{"readme":"README.MD","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null,"zenodo":null}},"created_at":"2023-06-02T14:16:08.000Z","updated_at":"2024-11-14T12:27:56.000Z","dependencies_parsed_at":"2023-09-26T16:46:05.552Z","dependency_job_id":"933aa7c2-ace6-4b1e-ab19-ed950c0f66cc","html_url":"https://github.com/ssbuild/rwkv_finetuning","commit_stats":null,"previous_names":["ssbuild/rwkv_finetuning"],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/ssbuild%2Frwkv_finetuning","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/ssbuild%2Frwkv_finetuning/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/ssbuild%2Frwkv_finetuning/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/ssbuild%2Frwkv_finetuning/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/ssbuild","download_url":"https://codeload.github.com/ssbuild/rwkv_finetuning/tar.gz/refs/heads/dev","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":250554361,"owners_count":21449594,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["lora","p-tuning-v2","qlora","rwkv","rwkv4"],"created_at":"2024-11-08T23:39:22.294Z","updated_at":"2025-04-24T03:27:39.428Z","avatar_url":"https://github.com/ssbuild.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"## update information\r\n   - [deep_training](https://github.com/ssbuild/deep_training)\r\n\r\n```text\r\n    2024-04-22 简化\r\n    2023-10-09 support accelerator trainer\r\n    2023-10-07 support colossalai trainer\r\n    2023-09-26 support transformers trainer\r\n    2023-08-02 增加 muti lora infer 例子, 手动升级 aigc_zoo , pip install -U git+https://github.com/ssbuild/deep_training.zoo.git --force-reinstall --no-deps\r\n    2023-07-16 support rwkv-4-world model training\r\n    2023-06-13 support resize_token_embeddings\r\n    2023-06-08 deep_training 0.1.10.post1  fix推理bug, 增加chat.py 最低版本要求0.1.10.post1\r\n    2023-06-07 deep_training 0.1.10.post0 fix setup for rwkv4  \r\n    2023-06-06 Initialize the rwkv finetuning \r\n```\r\n\r\n## install\r\n  - pip install -U -r requirements.txt\r\n  - 如果无法安装，可以切换官方源 pip install -i https://pypi.org/simple -U -r requirements.txt\r\n\r\n\r\n## weigtht select one is suitable for you !\r\n\r\n- [rwkv](https://huggingface.co/BlinkDL) \r\n- 使用 tools/convert_rwkv4.py 转换权重，配置及tokenizer\r\n\r\n\r\n\r\n## data sample\r\n- [open_data 不定时更新数据集](https://github.com/ssbuild/open_data)\r\n   \r\n单条数据示例\r\n```text\r\nrole one of user system function assistant\r\nq question optional\r\na answer   must\r\n\r\n```\r\n\r\n```json\r\n{\"id\": 1, \"paragraph\": [{\"q\": \"从南京到上海的路线\", \"a\": [\"你好，南京到上海的路线如下：\", \"1. 南京到上海，可以乘坐南京地铁1号线，在南京站乘坐轨道交通1号线。\", \"2. 南京到浦东机场，可以搭乘上海地铁1号，在陆家嘴站乘坐地铁1线，在浦东国际机场站乘坐机场快线，前往上海浦东国际机场。\", \"3. 上海到南京，可以换乘上海地铁2号线，从南京站换乘地铁2线，再从南京南站换乘地铁1路，然后到达上海站\"]}]}\r\n{\"id\": 2, \"paragraph\": [{\"q\": \"写一个诗歌，关于冬天\", \"a\": [\"冬夜寂静冷，\", \"云在天边飘，\", \"冰封白雪上， \", \"寒冷像一场雪。\", \" \", \"雪花融化成冰，\", \"像那雪花飘洒，\", \"在寒冷的冬天，\", \"感受春天的喜悦。\", \" 冬日里，\", \"风雪渐消，\", \"一片寂静，\", \"把快乐和温暖带回家。\"]}]}\r\n{\"id\": 3, \"paragraph\": [{\"q\": \"晚上睡不着应该怎么办\", \"a\": [\"晚上睡不着可能会让你感到焦虑或不舒服,但以下是一些可以帮助你入睡的方法:\", \"\", \"1. 制定规律的睡眠时间表:保持规律的睡眠时间表可以帮助你建立健康的睡眠习惯,使你更容易入睡。尽量在每天的相同时间上床,并在同一时间起床。\", \"2. 创造一个舒适的睡眠环境:确保睡眠环境舒适,安静,黑暗且温度适宜。可以使用舒适的床上用品,并保持房间通风。\", \"3. 放松身心:在睡前做些放松的活动,例如泡个热水澡,听些轻柔的音乐,阅读一些有趣的书籍等,有助于缓解紧张和焦虑,使你更容易入睡。\", \"4. 避免饮用含有咖啡因的饮料:咖啡因是一种刺激性物质,会影响你的睡眠质量。尽量避免在睡前饮用含有咖啡因的饮料,例如咖啡,茶和可乐。\", \"5. 避免在床上做与睡眠无关的事情:在床上做些与睡眠无关的事情,例如看电影,玩游戏或工作等,可能会干扰你的睡眠。\", \"6. 尝试呼吸技巧:深呼吸是一种放松技巧,可以帮助你缓解紧张和焦虑,使你更容易入睡。试着慢慢吸气,保持几秒钟,然后缓慢呼气。\", \"\", \"如果这些方法无法帮助你入睡,你可以考虑咨询医生或睡眠专家,寻求进一步的建议。\"]}]}\r\n```\r\n\r\n或者\r\n\r\n```json\r\n{\"id\": 1, \"conversations\": [{\"from\": \"user\", \"value\": \"从南京到上海的路线\"}, {\"from\": \"assistant\", \"value\": [\"你好，南京到上海的路线如下：\", \"1. 南京到上海，可以乘坐南京地铁1号线，在南京站乘坐轨道交通1号线。\", \"2. 南京到浦东机场，可以搭乘上海地铁1号，在陆家嘴站乘坐地铁1线，在浦东国际机场站乘坐机场快线，前往上海浦东国际机场。\", \"3. 上海到南京，可以换乘上海地铁2号线，从南京站换乘地铁2线，再从南京南站换乘地铁1路，然后到达上海站\"]}]}\r\n{\"id\": 2, \"conversations\": [{\"from\": \"user\", \"value\": \"写一个诗歌，关于冬天\"}, {\"from\": \"assistant\", \"value\": [\"冬夜寂静冷，\", \"云在天边飘，\", \"冰封白雪上， \", \"寒冷像一场雪。\", \" \", \"雪花融化成冰，\", \"像那雪花飘洒，\", \"在寒冷的冬天，\", \"感受春天的喜悦。\", \" 冬日里，\", \"风雪渐消，\", \"一片寂静，\", \"把快乐和温暖带回家。\"]}]}\r\n{\"id\": 3, \"conversations\": [{\"from\": \"user\", \"value\": \"晚上睡不着应该怎么办\"}, {\"from\": \"assistant\", \"value\": [\"晚上睡不着可能会让你感到焦虑或不舒服,但以下是一些可以帮助你入睡的方法:\", \"\", \"1. 制定规律的睡眠时间表:保持规律的睡眠时间表可以帮助你建立健康的睡眠习惯,使你更容易入睡。尽量在每天的相同时间上床,并在同一时间起床。\", \"2. 创造一个舒适的睡眠环境:确保睡眠环境舒适,安静,黑暗且温度适宜。可以使用舒适的床上用品,并保持房间通风。\", \"3. 放松身心:在睡前做些放松的活动,例如泡个热水澡,听些轻柔的音乐,阅读一些有趣的书籍等,有助于缓解紧张和焦虑,使你更容易入睡。\", \"4. 避免饮用含有咖啡因的饮料:咖啡因是一种刺激性物质,会影响你的睡眠质量。尽量避免在睡前饮用含有咖啡因的饮料,例如咖啡,茶和可乐。\", \"5. 避免在床上做与睡眠无关的事情:在床上做些与睡眠无关的事情,例如看电影,玩游戏或工作等,可能会干扰你的睡眠。\", \"6. 尝试呼吸技巧:深呼吸是一种放松技巧,可以帮助你缓解紧张和焦虑,使你更容易入睡。试着慢慢吸气,保持几秒钟,然后缓慢呼气。\", \"\", \"如果这些方法无法帮助你入睡,你可以考虑咨询医生或睡眠专家,寻求进一步的建议。\"]}]}\r\n```\r\n\r\n\r\n## infer\r\n    # infer_finetuning.py 推理微调模型\r\n    # infer_lora_finetuning.py 推理微调模型\r\n    # infer_ptuning.py 推理p-tuning-v2微调模型\r\n     python infer_finetuning.py\r\n\r\n\r\npython chat.py\r\n![chat.py](tools/1.png)\r\n\r\n\r\npython infer.py\r\n![infer.py](tools/2.png)\r\n\r\n\r\n\r\n\r\n\r\n## training\r\n```text\r\n# 制作数据\r\ncd scripts\r\nbash train_full.sh -m dataset \r\nor\r\nbash train_lora.sh -m dataset \r\nor\r\nbash train_ptv2.sh -m dataset \r\n\r\n注: num_process_worker 为多进程制作数据 ， 如果数据量较大 ， 适当调大至cpu数量\r\ndataHelper.make_dataset_with_args(data_args.train_file,mixed_data=False, shuffle=True,mode='train',num_process_worker=0)\r\n\r\n# 全参数训练 \r\n    bash train_full.sh -m train \r\n    \r\n# lora adalora ia3 \r\n    bash train_lora.sh -m train \r\n    \r\n# ptv2\r\n    bash train_ptv2.sh -m train \r\n```\r\n   \r\n## 训练参数\r\n[训练参数](args.MD)\r\n\r\n## 友情链接\r\n\r\n- [pytorch-task-example](https://github.com/ssbuild/pytorch-task-example)\r\n- [moss_finetuning](https://github.com/ssbuild/moss_finetuning)\r\n- [chatglm_finetuning](https://github.com/ssbuild/chatglm_finetuning)\r\n- [chatglm2_finetuning](https://github.com/ssbuild/chatglm2_finetuning)\r\n- [chatglm3_finetuning](https://github.com/ssbuild/chatglm3_finetuning)\r\n- [t5_finetuning](https://github.com/ssbuild/t5_finetuning)\r\n- [llm_finetuning](https://github.com/ssbuild/llm_finetuning)\r\n- [llm_rlhf](https://github.com/ssbuild/llm_rlhf)\r\n- [chatglm_rlhf](https://github.com/ssbuild/chatglm_rlhf)\r\n- [t5_rlhf](https://github.com/ssbuild/t5_rlhf)\r\n- [rwkv_finetuning](https://github.com/ssbuild/rwkv_finetuning)\r\n- [baichuan_finetuning](https://github.com/ssbuild/baichuan_finetuning)\r\n- [xverse_finetuning](https://github.com/ssbuild/xverse_finetuning)\r\n- [internlm_finetuning](https://github.com/ssbuild/internlm_finetuning)\r\n- [qwen_finetuning](https://github.com/ssbuild/qwen_finetuning)\r\n- [skywork_finetuning](https://github.com/ssbuild/skywork_finetuning)\r\n- [bluelm_finetuning](https://github.com/ssbuild/bluelm_finetuning)\r\n- [yi_finetuning](https://github.com/ssbuild/yi_finetuning)\r\n\r\n## \r\n    纯粹而干净的代码\r\n\r\n## Star History\r\n\r\n[![Star History Chart](https://api.star-history.com/svg?repos=ssbuild/rwkv_finetuning\u0026type=Date)](https://star-history.com/#ssbuild/rwkv_finetuning\u0026Date)","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fssbuild%2Frwkv_finetuning","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fssbuild%2Frwkv_finetuning","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fssbuild%2Frwkv_finetuning/lists"}