diff --git a/.DS_Store b/.DS_Store deleted file mode 100644 index d332226..0000000 Binary files a/.DS_Store and /dev/null differ diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..f4062ba --- /dev/null +++ b/requirements.txt @@ -0,0 +1 @@ +diffusers[torch] diff --git a/stable_diffusion.py b/stable_diffusion.py index c6137cb..5c4b59b 100644 --- a/stable_diffusion.py +++ b/stable_diffusion.py @@ -1,9 +1,7 @@ -import logging import torch from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler from towhee.operator import PyOperator -log = logging.getLogger(PyOperator) class StableDiffusion(PyOperator): def __init__(self,