diff --git a/README.md b/README.md index 759318a..1e7a3be 100644 --- a/README.md +++ b/README.md @@ -50,7 +50,7 @@ | InternLM2_7B_chat | 全量微调 | [internlm2_chat_7b_full.py](./xtuner_config/internlm2_chat_7b_full.py) | | InternLM2_7B_base | QLORA | [internlm2_7b_base_qlora_e10_M_1e4_32_64.py](./xtuner_config/internlm2_7b_base_qlora_e10_M_1e4_32_64.py) | | InternLM2_1_8B_chat | 全量微调 | [internlm2_1_8b_full_alpaca_e3.py](./xtuner_config/internlm2_1_8b_full_alpaca_e3.py) | -| InternLM2_20B_chat | LORA | | +| InternLM2_20B_chat | LORA |[internlm2_20b_chat_lora_alpaca_e3.py](./xtuner_config/internlm2_20b_chat_lora_alpaca_e3.py)| | Qwen_7b_chat | QLORA | [qwen_7b_chat_qlora_e3.py](./xtuner_config/qwen_7b_chat_qlora_e3.py) | | Qwen1_5-0_5B-Chat | 全量微调 | [qwen1_5_0_5_B_full.py](./xtuner_config/qwen1_5_0_5_B_full.py) | | Baichuan2_13B_chat | QLORA | [baichuan2_13b_chat_qlora_alpaca_e3.py](./xtuner_config/baichuan2_13b_chat_qlora_alpaca_e3.py) | diff --git a/README_EN.md b/README_EN.md index 4d6c396..8bd4562 100644 --- a/README_EN.md +++ b/README_EN.md @@ -52,7 +52,7 @@ | InternLM2_7B_chat | full fine-tuning | [internlm2_chat_7b_full.py](./xtuner_config/internlm2_chat_7b_full.py) | | InternLM2_7B_base | QLORA | [internlm2_7b_base_qlora_e10_M_1e4_32_64.py](./xtuner_config/internlm2_7b_base_qlora_e10_M_1e4_32_64.py) | | InternLM2_1_8B_chat | full fine-tuning | [internlm2_1_8b_full_alpaca_e3.py](./xtuner_config/internlm2_1_8b_full_alpaca_e3.py) | -| InternLM2_20B_chat | LORA | | +| InternLM2_20B_chat | LORA |[internlm2_20b_chat_lora_alpaca_e3.py](./xtuner_config/internlm2_20b_chat_lora_alpaca_e3.py)| | Qwen_7b_chat | QLORA | [qwen_7b_chat_qlora_e3.py](./xtuner_config/qwen_7b_chat_qlora_e3.py) | | Qwen1_5-0_5B-Chat | full fine-tuning | [qwen1_5_0_5_B_full.py](./xtuner_config/qwen1_5_0_5_B_full.py) | | Baichuan2_13B_chat | QLORA | [baichuan2_13b_chat_qlora_alpaca_e3.py](./xtuner_config/baichuan2_13b_chat_qlora_alpaca_e3.py) | diff --git a/app.py b/app.py index 6a650f3..3b24ce4 100644 --- a/app.py +++ b/app.py @@ -1,3 +1,14 @@ import os -os.system('streamlit run web_internlm2.py --server.address=0.0.0.0 --server.port 7860') -#os.system('streamlit run web_demo-aiwei.py --server.address=0.0.0.0 --server.port 7860') + +#model = "EmoLLM_aiwei" +model = "EmoLLM_Model" + +if model == "EmoLLM_aiwei": + os.system("python download_model.py ajupyter/EmoLLM_aiwei") + os.system('streamlit run web_demo-aiwei.py --server.address=0.0.0.0 --server.port 7860') +elif model == "EmoLLM_Model": + os.system("python download_model.py jujimeizuo/EmoLLM_Model") + os.system('streamlit run web_internlm2.py --server.address=0.0.0.0 --server.port 7860') +else: + print("Please select one model") +