wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-26 22:30:19 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Updates for better compatibility with DeepFloyd

Browse Source
This commit is contained in:
Benjamin Berman 2023-05-22 17:36:23 -07:00 committed by Benjamin Berman
parent 1cef5474d1
commit 550788eaca
3 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/utils.py
View File

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

13
requirements.txt
View File

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

1
setup.py
View File

@ -88,6 +88,7 @@ def _is_amd() -> bool:
def dependencies() -> [str]: def dependencies() -> [str]:
_dependencies = open(os.path.join(os.path.dirname(__file__), "requirements.txt")).readlines() _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] _alternative_indices = [amd_torch_index, nvidia_torch_index, cpu_torch_index_nightlies]
session = PipSession() session = PipSession()