sd

README.md

@@ -59,9 +59,11 @@ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float
 
 #pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 
-prompt = "an orange cat"/“a corgi”
+prompt = "an orange cat"
 
-image = pipe(prompt).images[0] image.save("orange_cat.png"/“corgi.png”)
+image = pipe(prompt).images[0] 
+
+image.save("orange_cat.png")
 ```
 .
 

Stable_diff_text2image.py

@@ -8,7 +8,7 @@ log = logging.getLogger(PyOperator)
 class StableDiffusion(PyOperator):
     def __init__(self,model_id='stabilityai/stable-diffusion-2-1',pipe='StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32'):
         self._model_id=model_id
-        self.pipe = pipe
+        self._pipe = pipe
 
     def __call__(self, prompt:str):
         # pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)