Update readme (#169)
							
								
								
									
										1
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						| @ -10,6 +10,7 @@ logs/ | |||||||
| *.jsonl | *.jsonl | ||||||
| *.json | *.json | ||||||
| *.txt | *.txt | ||||||
|  | localFile/ | ||||||
| # ./generate_data/*.josnl | # ./generate_data/*.josnl | ||||||
| # ./generate_data/*/*/*.josnl | # ./generate_data/*/*/*.josnl | ||||||
| 
 | 
 | ||||||
|  | |||||||
							
								
								
									
										62
									
								
								README.md
									
									
									
									
									
								
							
							
						
						| @ -6,7 +6,7 @@ | |||||||
| 
 | 
 | ||||||
| <p align="center"> | <p align="center"> | ||||||
|   <a href="https://github.com/SmartFlowAI/EmoLLM/"> |   <a href="https://github.com/SmartFlowAI/EmoLLM/"> | ||||||
|     <img src="assets/logo.jpeg" alt="Logo" width="30%"> |     <img src="assets/EmoLLM_transparent.png" alt="Logo" width="50%"> | ||||||
|   </a> |   </a> | ||||||
| 
 | 
 | ||||||
| <div align="center"> | <div align="center"> | ||||||
| @ -44,20 +44,20 @@ | |||||||
| 
 | 
 | ||||||
| <div align="center"> | <div align="center"> | ||||||
| 
 | 
 | ||||||
| |         模型          |    类型    | | |         模型          |   类型   | 链接  | | ||||||
| | :-------------------: | :--------: | | | :-------------------: | :------: | :---: | | ||||||
| |   InternLM2_7B_chat   |   QLORA   | | |   InternLM2_7B_chat   |  QLORA   |       | | ||||||
| |   InternLM2_7B_chat   |  全量微调  | | |   InternLM2_7B_chat   | 全量微调 |       | | ||||||
| |   InternLM2_7B_base   |   QLORA   | | |   InternLM2_7B_base   |  QLORA   |       | | ||||||
| |  InternLM2_1_8B_chat  |  全量微调  | | |  InternLM2_1_8B_chat  | 全量微调 |       | | ||||||
| |  InternLM2_20B_chat   |   LORA    | | |  InternLM2_20B_chat   |   LORA   |       | | ||||||
| |     Qwen_7b_chat      |   QLORA   | | |     Qwen_7b_chat      |  QLORA   |       | | ||||||
| |   Qwen1_5-0_5B-Chat   |  全量微调  | | |   Qwen1_5-0_5B-Chat   | 全量微调 |       | | ||||||
| |  Baichuan2_13B_chat   |   QLORA   | | |  Baichuan2_13B_chat   |  QLORA   |       | | ||||||
| |      ChatGLM3_6B      |   LORA    | | |      ChatGLM3_6B      |   LORA   |       | | ||||||
| | DeepSeek MoE_16B_chat |   QLORA   | | | DeepSeek MoE_16B_chat |  QLORA   |       | | ||||||
| | Mixtral 8x7B_instruct |   QLORA   | | | Mixtral 8x7B_instruct |  QLORA   |       | | ||||||
| |          ……           |    ……     | | |          ……           |    ……    |  ……   | | ||||||
| 
 | 
 | ||||||
| </div> | </div> | ||||||
| 
 | 
 | ||||||
| @ -76,6 +76,34 @@ | |||||||
| - 预防和干预措施:心理健康大模型还包括预防心理问题和促进心理健康的策略,如心理教育、心理咨询、心理治疗和社会支持系统。 | - 预防和干预措施:心理健康大模型还包括预防心理问题和促进心理健康的策略,如心理教育、心理咨询、心理治疗和社会支持系统。 | ||||||
| - 评估和诊断工具:为了有效促进心理健康,需要有科学的工具来评估个体的心理状态,以及诊断可能存在的心理问题。 | - 评估和诊断工具:为了有效促进心理健康,需要有科学的工具来评估个体的心理状态,以及诊断可能存在的心理问题。 | ||||||
|    |    | ||||||
|  | 
 | ||||||
|  | <table> | ||||||
|  |   <tr> | ||||||
|  |     <td align="center"> | ||||||
|  |       <a href="#"> | ||||||
|  |         <img src="assets/aiwei_demo1.gif" width="80%;" hight="80%;" alt="placeholder-image"> | ||||||
|  |       </a> | ||||||
|  |     </td> | ||||||
|  |     <td align="center"> | ||||||
|  |       <a href="#"> | ||||||
|  |         <img src="assets/aiwei_demo2.gif" width="80%;" hight="80%;" alt="placeholder-image"> | ||||||
|  |       </a> | ||||||
|  |     </td> | ||||||
|  |   </tr> | ||||||
|  |   <tr> | ||||||
|  |     <td align="center"> | ||||||
|  |       <a href="#"> | ||||||
|  |         <img src="assets/aiwei_demo3.gif" width="80%;" hight="80%;" alt="placeholder-image"> | ||||||
|  |       </a> | ||||||
|  |     </td> | ||||||
|  |     <td align="center"> | ||||||
|  |       <a href="#"> | ||||||
|  |         <img src="assets/aiwei_demo4.gif" width="80%;" hight="80%;" alt="placeholder-image"> | ||||||
|  |       </a> | ||||||
|  |     </td> | ||||||
|  |   </tr> | ||||||
|  | </table> | ||||||
|  | 
 | ||||||
| ### 🎇最近更新 | ### 🎇最近更新 | ||||||
| - 【2024.4.2】在 Huggingface 上传[老母亲心理咨询师](https://huggingface.co/brycewang2018/EmoLLM-mother/tree/main) | - 【2024.4.2】在 Huggingface 上传[老母亲心理咨询师](https://huggingface.co/brycewang2018/EmoLLM-mother/tree/main) | ||||||
| - 【2024.3.25】在百度飞桨平台发布[爹系男友心理咨询师](https://aistudio.baidu.com/community/app/68787) | - 【2024.3.25】在百度飞桨平台发布[爹系男友心理咨询师](https://aistudio.baidu.com/community/app/68787) | ||||||
| @ -117,7 +145,7 @@ | |||||||
|    |    | ||||||
| <p align="center"> | <p align="center"> | ||||||
|    <a href="https://github.com/SmartFlowAI/EmoLLM/"> |    <a href="https://github.com/SmartFlowAI/EmoLLM/"> | ||||||
|     <img src="assets/Shusheng.jpg" alt="浦语挑战赛创新创意奖"> |     <img src="assets/Shusheng.png" alt="浦语挑战赛创新创意奖"> | ||||||
| </p> | </p> | ||||||
| 
 | 
 | ||||||
| - 项目荣获公众号**NLP工程化**[推文宣传](https://mp.weixin.qq.com/s/78lrRl2tlXEKUfElnkVx4A) | - 项目荣获公众号**NLP工程化**[推文宣传](https://mp.weixin.qq.com/s/78lrRl2tlXEKUfElnkVx4A) | ||||||
| @ -226,7 +254,7 @@ git clone https://github.com/SmartFlowAI/EmoLLM.git | |||||||
| ### 作者(排名不分先后) | ### 作者(排名不分先后) | ||||||
| 
 | 
 | ||||||
| |                            用户名                             |                     学校/组织                      |                                 备注                                 |                     贡献                      | | |                            用户名                             |                     学校/组织                      |                                 备注                                 |                     贡献                      | | ||||||
| |:--------------------------------------------------------------------:|:--------------------------------------------------:| :-------------------: |:---------:| | | :-----------------------------------------------------------: | :------------------------------------------------: | :------------------------------------------------------------------: | :-------------------------------------------: | | ||||||
| |            [aJupyter](https://github.com/aJupyter)            |                  南开大学在读硕士                  |                            DataWhale成员                             |                  项目发起人                   | | |            [aJupyter](https://github.com/aJupyter)            |                  南开大学在读硕士                  |                            DataWhale成员                             |                  项目发起人                   | | ||||||
| |            [MING-ZCH](https://github.com/MING-ZCH)            |               华中科技大学在读本科生               |                       LLM x Psychology 研究者                        |                项目联合负责人                 | | |            [MING-ZCH](https://github.com/MING-ZCH)            |               华中科技大学在读本科生               |                       LLM x Psychology 研究者                        |                项目联合负责人                 | | ||||||
| |          [jujimeizuo](https://github.com/jujimeizuo)          |                  江南大学在读硕士                  |                                                                      |                                               | | |          [jujimeizuo](https://github.com/jujimeizuo)          |                  江南大学在读硕士                  |                                                                      |                                               | | ||||||
|  | |||||||
							
								
								
									
										72
									
								
								README_EN.md
									
									
									
									
									
								
							
							
						
						| @ -6,7 +6,7 @@ | |||||||
| 
 | 
 | ||||||
| <p align="center"> | <p align="center"> | ||||||
|   <a href="https://github.com/SmartFlowAI/EmoLLM/"> |   <a href="https://github.com/SmartFlowAI/EmoLLM/"> | ||||||
|     <img src="assets/logo.jpeg" alt="Logo" width="30%"> |     <img src="assets/EmoLLM_transparent.png" alt="Logo" width="50%"> | ||||||
|   </a> |   </a> | ||||||
| 
 | 
 | ||||||
| <div align="center"> | <div align="center"> | ||||||
| @ -46,20 +46,20 @@ | |||||||
| 
 | 
 | ||||||
| <div align="center"> | <div align="center"> | ||||||
| 
 | 
 | ||||||
| |         Model          |   Type   | | |         Model         |       Type       | link  | | ||||||
| | :-------------------: | :------: | | | :-------------------: | :--------------: | :---: | | ||||||
| |   InternLM2_7B_chat   |  QLORA   | | |   InternLM2_7B_chat   |      QLORA       |       | | ||||||
| |   InternLM2_7B_chat   | full fine-tuning  | | |   InternLM2_7B_chat   | full fine-tuning |       | | ||||||
| |   InternLM2_7B_base   |   QLORA   | | |   InternLM2_7B_base   |      QLORA       |       | | ||||||
| |  InternLM2_1_8B_chat  | full fine-tuning  | | |  InternLM2_1_8B_chat  | full fine-tuning |       | | ||||||
| |  InternLM2_20B_chat  | LORA    | | |  InternLM2_20B_chat   |       LORA       |       | | ||||||
| |     Qwen_7b_chat      |   QLORA   | | |     Qwen_7b_chat      |      QLORA       |       | | ||||||
| |   Qwen1_5-0_5B-Chat   | full fine-tuning  | | |   Qwen1_5-0_5B-Chat   | full fine-tuning |       | | ||||||
| |  Baichuan2_13B_chat   |   QLORA   | | |  Baichuan2_13B_chat   |      QLORA       |       | | ||||||
| |      ChatGLM3_6B      |   LORA    | | |      ChatGLM3_6B      |       LORA       |       | | ||||||
| | DeepSeek MoE_16B_chat |  QLORA   | | | DeepSeek MoE_16B_chat |      QLORA       |       | | ||||||
| | Mixtral 8x7B_instruct |  QLORA   | | | Mixtral 8x7B_instruct |      QLORA       |       | | ||||||
| |          ……           |    ……    | | |          ……           |        ……        |  ……   | | ||||||
| 
 | 
 | ||||||
| </div> | </div> | ||||||
| 
 | 
 | ||||||
| @ -78,6 +78,34 @@ The Model aims to fully understand and promote the mental health of individuals, | |||||||
| - Prevention and intervention measures: The Mental Health Grand Model also includes strategies for preventing psychological issues and promoting mental health, such as psychological education, counseling, therapy, and social support systems. | - Prevention and intervention measures: The Mental Health Grand Model also includes strategies for preventing psychological issues and promoting mental health, such as psychological education, counseling, therapy, and social support systems. | ||||||
| - Assessment and diagnostic tools: Effective promotion of mental health requires scientific tools to assess individuals' psychological states and diagnose potential psychological issues. | - Assessment and diagnostic tools: Effective promotion of mental health requires scientific tools to assess individuals' psychological states and diagnose potential psychological issues. | ||||||
|   |   | ||||||
|  | 
 | ||||||
|  | <table> | ||||||
|  |   <tr> | ||||||
|  |     <td align="center"> | ||||||
|  |       <a href="#"> | ||||||
|  |         <img src="assets/aiwei_demo1.gif" width="80%;" hight="80%;" alt="placeholder-image"> | ||||||
|  |       </a> | ||||||
|  |     </td> | ||||||
|  |     <td align="center"> | ||||||
|  |       <a href="#"> | ||||||
|  |         <img src="assets/aiwei_demo2.gif" width="80%;" hight="80%;" alt="placeholder-image"> | ||||||
|  |       </a> | ||||||
|  |     </td> | ||||||
|  |   </tr> | ||||||
|  |   <tr> | ||||||
|  |     <td align="center"> | ||||||
|  |       <a href="#"> | ||||||
|  |         <img src="assets/aiwei_demo3.gif" width="80%;" hight="80%;" alt="placeholder-image"> | ||||||
|  |       </a> | ||||||
|  |     </td> | ||||||
|  |     <td align="center"> | ||||||
|  |       <a href="#"> | ||||||
|  |         <img src="assets/aiwei_demo4.gif" width="80%;" hight="80%;" alt="placeholder-image"> | ||||||
|  |       </a> | ||||||
|  |     </td> | ||||||
|  |   </tr> | ||||||
|  | </table> | ||||||
|  | 
 | ||||||
| ### Recent Updates | ### Recent Updates | ||||||
| - 【2024.3.25】 [Mother-like Therapist] is released on Huggingface (https://huggingface.co/brycewang2018/EmoLLM-mother/tree/main) | - 【2024.3.25】 [Mother-like Therapist] is released on Huggingface (https://huggingface.co/brycewang2018/EmoLLM-mother/tree/main) | ||||||
| - 【2024.3.25】 [Daddy-like Boy-Friend] is released on Baidu Paddle-Paddle AI Studio Platform (https://aistudio.baidu.com/community/app/68787) | - 【2024.3.25】 [Daddy-like Boy-Friend] is released on Baidu Paddle-Paddle AI Studio Platform (https://aistudio.baidu.com/community/app/68787) | ||||||
| @ -121,7 +149,7 @@ The Model aims to fully understand and promote the mental health of individuals, | |||||||
| 
 | 
 | ||||||
| <p align="center"> | <p align="center"> | ||||||
|    <a href="https://github.com/SmartFlowAI/EmoLLM/"> |    <a href="https://github.com/SmartFlowAI/EmoLLM/"> | ||||||
|     <img src="assets/Shusheng.jpg" alt="Challenge Innovation and Creativity Award"> |     <img src="assets/Shusheng.png" alt="Challenge Innovation and Creativity Award"> | ||||||
| </p> | </p> | ||||||
| 
 | 
 | ||||||
| 
 | 
 | ||||||
| @ -138,7 +166,7 @@ The Model aims to fully understand and promote the mental health of individuals, | |||||||
| 
 | 
 | ||||||
| - [EmoLLM - Large Language Model for Mental Health](#emollm---large-language-model-for-mental-health) | - [EmoLLM - Large Language Model for Mental Health](#emollm---large-language-model-for-mental-health) | ||||||
|     - [Recent Updates](#recent-updates) |     - [Recent Updates](#recent-updates) | ||||||
|     - [Honor](#honor) |     - [Honors](#honors) | ||||||
|     - [Roadmap](#roadmap) |     - [Roadmap](#roadmap) | ||||||
|   - [Contents](#contents) |   - [Contents](#contents) | ||||||
|           - [Pre-development Configuration Requirements.](#pre-development-configuration-requirements) |           - [Pre-development Configuration Requirements.](#pre-development-configuration-requirements) | ||||||
| @ -192,18 +220,18 @@ git clone https://github.com/SmartFlowAI/EmoLLM.git | |||||||
| 
 | 
 | ||||||
| ### Data Construction | ### Data Construction | ||||||
| 
 | 
 | ||||||
| - Please read the [Data Construction Guide ](generate_data/tutorial.md)for reference. | - Please read the [Data Construction Guide ](generate_data/tutorial_EN.md)for reference. | ||||||
| 
 | 
 | ||||||
| - The dataset used for this fine-tuning can be found at [datasets](datasets/data.json) | - The dataset used for this fine-tuning can be found at [datasets](datasets/data.json) | ||||||
| 
 | 
 | ||||||
| ### Fine-tuning Guide | ### Fine-tuning Guide | ||||||
| 
 | 
 | ||||||
| For details, see the [fine-tuning guide](xtuner_config/README.md) | For details, see the [fine-tuning guide](xtuner_config/README_EN.md) | ||||||
| 
 | 
 | ||||||
| ### Deployment Guide | ### Deployment Guide | ||||||
| 
 | 
 | ||||||
| - Demo deployment: see [deployment guide](./demo/README.md) for details. | - Demo deployment: see [deployment guide](./demo/README_EN.md) for details. | ||||||
| - Quantitative deployment based on [LMDeploy](https://github.com/InternLM/lmdeploy/): see [deploy](./deploy/lmdeploy.md) | - Quantitative deployment based on [LMDeploy](https://github.com/InternLM/lmdeploy/): see [deploy](./deploy/lmdeploy_EN.md) | ||||||
| 
 | 
 | ||||||
| ### RAG (Retrieval Augmented Generation) Pipeline | ### RAG (Retrieval Augmented Generation) Pipeline | ||||||
| 
 | 
 | ||||||
| @ -241,7 +269,7 @@ This project uses Git for version control. You can see the currently available v | |||||||
| ### Authors (in no particular order) | ### Authors (in no particular order) | ||||||
| 
 | 
 | ||||||
| |                           Username                            |                         School/Organization                          |                                  Remarks                                  |                                     Contributions                                     | | |                           Username                            |                         School/Organization                          |                                  Remarks                                  |                                     Contributions                                     | | ||||||
| |:-----------------------------------------------------------------------:|:--------------------------------------------------------------------:| :------------------: |:----------------------------------:| | | :-----------------------------------------------------------: | :------------------------------------------------------------------: | :-----------------------------------------------------------------------: | :-----------------------------------------------------------------------------------: | | ||||||
| |            [aJupyter](https://github.com/aJupyter)            |                 Nankai University, Master's student                  |                             DataWhale member                              |                                   Project initiator                                   | | |            [aJupyter](https://github.com/aJupyter)            |                 Nankai University, Master's student                  |                             DataWhale member                              |                                   Project initiator                                   | | ||||||
| |            [MING-ZCH](https://github.com/MING-ZCH)            | Huazhong University of Science and Technology, Undergraduate student |                        LLM X Psychology researcher                        |                                   Project co-leader                                   | | |            [MING-ZCH](https://github.com/MING-ZCH)            | Huazhong University of Science and Technology, Undergraduate student |                        LLM X Psychology researcher                        |                                   Project co-leader                                   | | ||||||
| |          [jujimeizuo](https://github.com/jujimeizuo)          |                Jiangnan University, Master's student                 |                                                                           |                                                                                       | | |          [jujimeizuo](https://github.com/jujimeizuo)          |                Jiangnan University, Master's student                 |                                                                           |                                                                                       | | ||||||
|  | |||||||
							
								
								
									
										
											BIN
										
									
								
								assets/EmoLLM_transparent.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 67 KiB | 
| Before Width: | Height: | Size: 120 KiB | 
							
								
								
									
										
											BIN
										
									
								
								assets/Shusheng.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 3.4 MiB | 
							
								
								
									
										
											BIN
										
									
								
								assets/aiwei_demo1.gif
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 1.7 MiB | 
							
								
								
									
										
											BIN
										
									
								
								assets/aiwei_demo2.gif
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 1.1 MiB | 
							
								
								
									
										
											BIN
										
									
								
								assets/aiwei_demo3.gif
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 1.8 MiB | 
							
								
								
									
										
											BIN
										
									
								
								assets/aiwei_demo4.gif
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 536 KiB | 
 xzw
						xzw