diff --git a/pyproject.toml b/pyproject.toml index 74506f6..a44441f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -15,7 +15,7 @@ PublisherId = "sonnybox" DisplayName = "ComfyUI-SuperNodes" Icon = "https://raw.githubusercontent.com/sonnybox/ComfyUI-SuperNodes/refs/heads/main/logo.png" includes = [] -dependencies = ["numpy","pillow",] +dependencies = ["numpy", "pillow", "torch", "diffusers", "matplotlib"] # "requires-comfyui" = ">=1.0.0" # ComfyUI version compatibility [tool.basedpyright] diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..2c84065 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,5 @@ +numpy +pillow +torch +diffusers +matplotlib \ No newline at end of file diff --git a/scheduler.py b/scheduler.py index 2d09b2d..ac280c5 100644 --- a/scheduler.py +++ b/scheduler.py @@ -4,7 +4,7 @@ import os import random import diffusers import torch -import folder_paths +import folder_paths # type: ignore import numpy as np import matplotlib.pyplot as plt import io