Updates for better compatibility with DeepFloyd

comfy/utils.py

@@ -1,3 +1,5 @@
+import os.path
+
 import torch
 import math
 import struct
@@ -379,3 +381,7 @@ class ProgressBar:
 
     def update(self, value):
         self.update_absolute(self.current + value)
+
+
+def get_project_root() -> str:
+    return os.path.join(os.path.dirname(__file__), "..")

requirements.txt

@@ -1,11 +1,11 @@
-torch>=1.13.0
-torchvision
-torchaudio
+torch==2.0.0
+torchaudio==2.0.0
+torchvision==0.15.1
 torchdiffeq>=0.2.3
 torchsde>=0.2.5
 einops>=0.6.0
 open-clip-torch>=2.16.0
-transformers>=4.25.1
+transformers>=4.29.1
 safetensors>=0.3.0
 pytorch-lightning>=2.0.0
 aiohttp>=3.8.4
@@ -24,4 +24,7 @@ python-dateutil>=2.8.2
 importlib_resources
 Pillow
 scipy
-tqdm
+tqdm
+diffusers>=0.16.1
+protobuf==3.20.3
+rembg[gpu]

setup.py

@@ -88,6 +88,7 @@ def _is_amd() -> bool:
 
 def dependencies() -> [str]:
     _dependencies = open(os.path.join(os.path.dirname(__file__), "requirements.txt")).readlines()
+    # todo: also add all plugin dependencies
     _alternative_indices = [amd_torch_index, nvidia_torch_index, cpu_torch_index_nightlies]
     session = PipSession()