diff --git a/README.md b/README.md index ab33da3..376d79b 100644 --- a/README.md +++ b/README.md @@ -469,7 +469,7 @@ git clone https://github.com/ali-vilab/ACE_plus.git Install the necessary packages with `pip`: ```bash cd ACE_plus -pip install -r requirements.txt +pip install -r repo_requirements.txt ``` ACE++ depends on FLUX.1-Fill-dev as its base model, which you can download from [![HuggingFace link](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Model-yellow)](https://huggingface.co/black-forest-labs/FLUX.1-Fill-dev). In order to run the inference code or Gradio demo normally, we have defined the relevant environment variables to specify the location of the model. diff --git a/repo_requirements.txt b/repo_requirements.txt new file mode 100644 index 0000000..d92654d --- /dev/null +++ b/repo_requirements.txt @@ -0,0 +1,8 @@ +huggingface_hub +diffusers +transformers +torch>=2.4.1 +xformers>=0.0.27.post2 +gradio>=4.44.1 +scepter +ms_swift diff --git a/requirements.txt b/requirements.txt index de32498..8b13789 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,8 +1 @@ -huggingface_hub -diffusers -transformers -torch>=2.4.1 -xformers>=0.0.27.post2 -gradio>=4.44.1 -scepter -ms_swift \ No newline at end of file + diff --git a/workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py b/workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py index df75f25..689bf5a 100644 --- a/workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py +++ b/workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py @@ -141,8 +141,9 @@ class AcePlusFFTProcessor: self.max_aspect_ratio = max_aspect_ratio self.max_seq_len = max_seq_len self.d = d - self.processor_cfg = self.load_yaml(os.path.join('custom_nodes/ComfyUI-ACE_Plus/', - 'config/ace_plus_fft_processor.yaml')) + current_dir = os.path.dirname(os.path.abspath(__file__)) + config_path = os.path.join(current_dir, 'config', 'ace_plus_fft_processor.yaml') + self.processor_cfg = self.load_yaml(config_path) self.task_list = {} for task in self.processor_cfg['PREPROCESSOR']: self.task_list[task['TYPE']] = task