From 7255b5c4bc79a6828830ceea32746e0dc00e966b Mon Sep 17 00:00:00 2001 From: LucipherDev Date: Thu, 9 Jan 2025 15:09:32 +0530 Subject: [PATCH] Changed text_encoders dir --- README.md | 4 ++-- install.py | 2 +- nodes.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 8bcaca1..9d1fc4e 100644 --- a/README.md +++ b/README.md @@ -29,13 +29,13 @@ python install.py ## Usage -**If you installed from the ComfyUI-Manager, all the necessary models should be automatically downloaded into the `models/tangoflux` folder.** +**If you installed from the ComfyUI-Manager, all the necessary models should be automatically downloaded into the `models/tangoflux` and `models/text_encoders` folders.** **Otherwise they will be automatically downloaded when the LoadAniDoc node is used for the first time.** **Manual Download:** - Download TangoFlux from [here](https://huggingface.co/declare-lab/TangoFlux/tree/main) and put everything in `models/tangoflux` (make sure to include the config files) -- Download text encoders from [here](https://huggingface.co/google/flan-t5-large/tree/main) and put everything in `models/tangoflux/text_encoders/google-flan-t5-large` +- Download text encoders from [here](https://huggingface.co/google/flan-t5-large/tree/main) and put everything in `models/text_encoders/google-flan-t5-large` (make sure to include the config files) The nodes can be found in "TangoFlux" category as `TangoFluxDocLoader`, `TangoFluxSampler`, `TangoFluxVAEDecodeAndPlay`. diff --git a/install.py b/install.py index 1a81b0f..ad7335b 100644 --- a/install.py +++ b/install.py @@ -13,7 +13,7 @@ download_models = True try: import folder_paths TANGOFLUX_DIR = os.path.join(folder_paths.models_dir, "tangoflux") - TEXT_ENCODER_DIR = os.path.join(TANGOFLUX_DIR, "text_encoders") + TEXT_ENCODER_DIR = os.path.join(folder_paths.models_dir, "text_encoders") except: download_models = False log.info("Not called by ComfyUI Manager. Models will not be downloaded") diff --git a/nodes.py b/nodes.py index f9d048a..f4090cf 100644 --- a/nodes.py +++ b/nodes.py @@ -25,7 +25,7 @@ folder_paths.folder_names_and_paths["tangoflux"] = ( current_paths, folder_paths.supported_pt_extensions, ) -TEXT_ENCODER_DIR = os.path.join(TANGOFLUX_DIR, "text_encoders") +TEXT_ENCODER_DIR = os.path.join(folder_paths.models_dir, "text_encoders") class TangoFluxLoader: @classmethod