Fix install

Author: LucipherDev

README.md

@@ -29,9 +29,9 @@ python install.py
 
 ## Usage
 
-**If you installed from the ComfyUI-Manager, all the necessary models should be automatically downloaded into the `models/tangoflux` and `models/text_encoders` folders.**
+**All the necessary models should be automatically downloaded when the TangoFluxLoader node is used for the first time.**
 
-**Otherwise they will be automatically downloaded when the TangoFluxLoader node is used for the first time.**
+**Models can also be downloaded using the `install.py` script**
 
 **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)

install.py

@@ -10,15 +10,19 @@
 
 download_models = True
 
+EXT_PATH = os.path.dirname(os.path.abspath(__file__))
+
 try:
+    folder_paths_path = os.path.abspath(os.path.join(EXT_PATH, "..", "..", "folder_paths.py"))
+
+    sys.path.append(os.path.dirname(folder_paths_path))
+
     import folder_paths
+    
     TANGOFLUX_DIR = os.path.join(folder_paths.models_dir, "tangoflux")
     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")
-
-EXT_PATH = os.path.dirname(os.path.abspath(__file__))
 
 try:
     log.info(f"Installing requirements")

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-tangoflux"
 description = "ComfyUI Custom Nodes for 'TangoFlux: Super Fast and Faithful Text to Audio Generation with Flow Matching'. This generates high-quality 44.1kHz audio up to 30 seconds using just a text prompt."
-version = "1.0.1"
+version = "1.0.2"
 license = {file = "LICENSE"}
 dependencies = ["torch", "torchaudio", "transformers", "diffusers", "accelerate", "tqdm"]
 

requirements.txt

@@ -1,7 +1,4 @@
-torch
 torchaudio
-transformers
 diffusers
 accelerate
-tqdm
-
+tqdm