Changes to be committed:
modified: .gitignore modified: README.md modified: README_EN.md new file: assets/EmoLLM_transparent.png deleted: assets/Shusheng.jpg new file: assets/Shusheng.png new file: assets/aiwei_demo1.gif new file: assets/aiwei_demo2.gif new file: assets/aiwei_demo3.gif new file: assets/aiwei_demo4.gif
1
.gitignore
vendored
@ -10,6 +10,7 @@ logs/
|
||||
*.jsonl
|
||||
*.json
|
||||
*.txt
|
||||
localFile/
|
||||
# ./generate_data/*.josnl
|
||||
# ./generate_data/*/*/*.josnl
|
||||
|
||||
|
50
README.md
@ -6,7 +6,7 @@
|
||||
|
||||
<p align="center">
|
||||
<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>
|
||||
|
||||
<div align="center">
|
||||
@ -44,20 +44,20 @@
|
||||
|
||||
<div align="center">
|
||||
|
||||
| 模型 | 类型 |
|
||||
| :-------------------: | :--------: |
|
||||
| InternLM2_7B_chat | QLORA |
|
||||
| InternLM2_7B_chat | 全量微调 |
|
||||
| InternLM2_7B_base | QLORA |
|
||||
| InternLM2_1_8B_chat | 全量微调 |
|
||||
| InternLM2_20B_chat | LORA |
|
||||
| Qwen_7b_chat | QLORA |
|
||||
| Qwen1_5-0_5B-Chat | 全量微调 |
|
||||
| Baichuan2_13B_chat | QLORA |
|
||||
| ChatGLM3_6B | LORA |
|
||||
| DeepSeek MoE_16B_chat | QLORA |
|
||||
| Mixtral 8x7B_instruct | QLORA |
|
||||
| …… | …… |
|
||||
| 模型 | 类型 | 链接 |
|
||||
| :-------------------: | :------: | :---: |
|
||||
| InternLM2_7B_chat | QLORA | |
|
||||
| InternLM2_7B_chat | 全量微调 | |
|
||||
| InternLM2_7B_base | QLORA | |
|
||||
| InternLM2_1_8B_chat | 全量微调 | |
|
||||
| InternLM2_20B_chat | LORA | |
|
||||
| Qwen_7b_chat | QLORA | |
|
||||
| Qwen1_5-0_5B-Chat | 全量微调 | |
|
||||
| Baichuan2_13B_chat | QLORA | |
|
||||
| ChatGLM3_6B | LORA | |
|
||||
| DeepSeek MoE_16B_chat | QLORA | |
|
||||
| Mixtral 8x7B_instruct | QLORA | |
|
||||
| …… | …… | …… |
|
||||
|
||||
</div>
|
||||
|
||||
@ -76,6 +76,22 @@
|
||||
- 预防和干预措施:心理健康大模型还包括预防心理问题和促进心理健康的策略,如心理教育、心理咨询、心理治疗和社会支持系统。
|
||||
- 评估和诊断工具:为了有效促进心理健康,需要有科学的工具来评估个体的心理状态,以及诊断可能存在的心理问题。
|
||||
|
||||
|
||||
<div style="display: grid; grid-template-columns: repeat(2, 1fr); grid-gap: 30px; margin: 50px;">
|
||||
<div style="background-color: transparent; padding: 10px;">
|
||||
<img class="image" src="assets/aiwei_demo1.gif" alt="艾薇演示1">
|
||||
</div>
|
||||
<div style="background-color: transparent; padding: 10px;">
|
||||
<img class="image" src="assets/aiwei_demo2.gif" alt="艾薇演示2">
|
||||
</div>
|
||||
<div style="background-color: transparent; padding: 10px;">
|
||||
<img class="image" src="assets/aiwei_demo3.gif" alt="艾薇演示3">
|
||||
</div>
|
||||
<div style="background-color: transparent; padding: 10px;">
|
||||
<img class="image" src="assets/aiwei_demo4.gif" alt="艾薇演示4">
|
||||
</div>
|
||||
</div>
|
||||
|
||||
### 🎇最近更新
|
||||
- 【2024.4.2】在 Huggingface 上传[老母亲心理咨询师](https://huggingface.co/brycewang2018/EmoLLM-mother/tree/main)
|
||||
- 【2024.3.25】在百度飞桨平台发布[爹系男友心理咨询师](https://aistudio.baidu.com/community/app/68787)
|
||||
@ -117,7 +133,7 @@
|
||||
|
||||
<p align="center">
|
||||
<a href="https://github.com/SmartFlowAI/EmoLLM/">
|
||||
<img src="assets/Shusheng.jpg" alt="浦语挑战赛创新创意奖">
|
||||
<img src="assets/Shusheng.png" alt="浦语挑战赛创新创意奖">
|
||||
</p>
|
||||
|
||||
- 项目荣获公众号**NLP工程化**[推文宣传](https://mp.weixin.qq.com/s/78lrRl2tlXEKUfElnkVx4A)
|
||||
@ -226,7 +242,7 @@ git clone https://github.com/SmartFlowAI/EmoLLM.git
|
||||
### 作者(排名不分先后)
|
||||
|
||||
| 用户名 | 学校/组织 | 备注 | 贡献 |
|
||||
|:--------------------------------------------------------------------:|:--------------------------------------------------:| :-------------------: |:---------:|
|
||||
| :-----------------------------------------------------------: | :------------------------------------------------: | :------------------------------------------------------------------: | :-------------------------------------------: |
|
||||
| [aJupyter](https://github.com/aJupyter) | 南开大学在读硕士 | DataWhale成员 | 项目发起人 |
|
||||
| [MING-ZCH](https://github.com/MING-ZCH) | 华中科技大学在读本科生 | LLM x Psychology 研究者 | 项目联合负责人 |
|
||||
| [jujimeizuo](https://github.com/jujimeizuo) | 江南大学在读硕士 | | |
|
||||
|
59
README_EN.md
@ -6,7 +6,7 @@
|
||||
|
||||
<p align="center">
|
||||
<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>
|
||||
|
||||
<div align="center">
|
||||
@ -46,20 +46,20 @@
|
||||
|
||||
<div align="center">
|
||||
|
||||
| Model | Type |
|
||||
| :-------------------: | :------: |
|
||||
| InternLM2_7B_chat | QLORA |
|
||||
| InternLM2_7B_chat | full fine-tuning |
|
||||
| InternLM2_7B_base | QLORA |
|
||||
| InternLM2_1_8B_chat | full fine-tuning |
|
||||
| InternLM2_20B_chat | LORA |
|
||||
| Qwen_7b_chat | QLORA |
|
||||
| Qwen1_5-0_5B-Chat | full fine-tuning |
|
||||
| Baichuan2_13B_chat | QLORA |
|
||||
| ChatGLM3_6B | LORA |
|
||||
| DeepSeek MoE_16B_chat | QLORA |
|
||||
| Mixtral 8x7B_instruct | QLORA |
|
||||
| …… | …… |
|
||||
| Model | Type | link |
|
||||
| :-------------------: | :--------------: | :---: |
|
||||
| InternLM2_7B_chat | QLORA | |
|
||||
| InternLM2_7B_chat | full fine-tuning | |
|
||||
| InternLM2_7B_base | QLORA | |
|
||||
| InternLM2_1_8B_chat | full fine-tuning | |
|
||||
| InternLM2_20B_chat | LORA | |
|
||||
| Qwen_7b_chat | QLORA | |
|
||||
| Qwen1_5-0_5B-Chat | full fine-tuning | |
|
||||
| Baichuan2_13B_chat | QLORA | |
|
||||
| ChatGLM3_6B | LORA | |
|
||||
| DeepSeek MoE_16B_chat | QLORA | |
|
||||
| Mixtral 8x7B_instruct | QLORA | |
|
||||
| …… | …… | …… |
|
||||
|
||||
</div>
|
||||
|
||||
@ -78,6 +78,21 @@ 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.
|
||||
- Assessment and diagnostic tools: Effective promotion of mental health requires scientific tools to assess individuals' psychological states and diagnose potential psychological issues.
|
||||
|
||||
<div style="display: grid; grid-template-columns: repeat(2, 1fr); grid-gap: 30px; margin: 50px;">
|
||||
<div style="background-color: transparent; padding: 10px;">
|
||||
<img class="image" src="assets/aiwei_demo1.gif" alt="艾薇演示1">
|
||||
</div>
|
||||
<div style="background-color: transparent; padding: 10px;">
|
||||
<img class="image" src="assets/aiwei_demo2.gif" alt="艾薇演示2">
|
||||
</div>
|
||||
<div style="background-color: transparent; padding: 10px;">
|
||||
<img class="image" src="assets/aiwei_demo3.gif" alt="艾薇演示3">
|
||||
</div>
|
||||
<div style="background-color: transparent; padding: 10px;">
|
||||
<img class="image" src="assets/aiwei_demo4.gif" alt="艾薇演示4">
|
||||
</div>
|
||||
</div>
|
||||
|
||||
### Recent Updates
|
||||
- 【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)
|
||||
@ -121,7 +136,7 @@ The Model aims to fully understand and promote the mental health of individuals,
|
||||
|
||||
<p align="center">
|
||||
<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>
|
||||
|
||||
|
||||
@ -138,7 +153,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)
|
||||
- [Recent Updates](#recent-updates)
|
||||
- [Honor](#honor)
|
||||
- [Honors](#honors)
|
||||
- [Roadmap](#roadmap)
|
||||
- [Contents](#contents)
|
||||
- [Pre-development Configuration Requirements.](#pre-development-configuration-requirements)
|
||||
@ -192,18 +207,18 @@ git clone https://github.com/SmartFlowAI/EmoLLM.git
|
||||
|
||||
### 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)
|
||||
|
||||
### 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
|
||||
|
||||
- Demo deployment: see [deployment guide](./demo/README.md) for details.
|
||||
- Quantitative deployment based on [LMDeploy](https://github.com/InternLM/lmdeploy/): see [deploy](./deploy/lmdeploy.md)
|
||||
- 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_EN.md)
|
||||
|
||||
### RAG (Retrieval Augmented Generation) Pipeline
|
||||
|
||||
@ -241,7 +256,7 @@ This project uses Git for version control. You can see the currently available v
|
||||
### Authors (in no particular order)
|
||||
|
||||
| Username | School/Organization | Remarks | Contributions |
|
||||
|:-----------------------------------------------------------------------:|:--------------------------------------------------------------------:| :------------------: |:----------------------------------:|
|
||||
| :-----------------------------------------------------------: | :------------------------------------------------------------------: | :-----------------------------------------------------------------------: | :-----------------------------------------------------------------------------------: |
|
||||
| [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 |
|
||||
| [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 |