From ff836cad5bcba241d8801824ee92267f3680c118 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=9A=93=E7=AB=A5?= Date: Thu, 16 Jan 2025 18:15:32 +0800 Subject: [PATCH] modify requirements --- README.md | 1 + requirements.txt | 3 ++- train_config/ace_plus_lora.yaml | 2 +- 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index b99d8f8..74f7e9e 100644 --- a/README.md +++ b/README.md @@ -38,6 +38,7 @@ ## 📢 News - [x] **[2025.01.06]** Release the code and models of ACE++. - [x] **[2025.01.07]** Release the demo on [HuggingFace](https://huggingface.co/spaces/scepter-studio/ACE-Plus). +- [x] **[2025.01.16]** Release the training code for lora. - [] **[ToDo]** Update Models. ## 🔥 ACE Models diff --git a/requirements.txt b/requirements.txt index 105fe57..de32498 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,4 +4,5 @@ transformers torch>=2.4.1 xformers>=0.0.27.post2 gradio>=4.44.1 -scepter \ No newline at end of file +scepter +ms_swift \ No newline at end of file diff --git a/train_config/ace_plus_lora.yaml b/train_config/ace_plus_lora.yaml index 73b0ed7..e47615b 100644 --- a/train_config/ace_plus_lora.yaml +++ b/train_config/ace_plus_lora.yaml @@ -274,4 +274,4 @@ SOLVER: EVAL_HOOKS: - NAME: ProbeDataHook PROB_INTERVAL: 50 - PRIORITY: 0 + PRIORITY: 0 \ No newline at end of file