Add requirements

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
1 year ago · 0ed039695d
3 changed files with 1 additions and 2 deletions
--- a/.DS_Store
+++ b/.DS_Store
--- a/requirements.txt
+++ b/requirements.txt
@ -0,0 +1 @@
+diffusers[torch]
--- 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,