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