diff --git a/README.md b/README.md index 3d057b1..3101075 100644 --- a/README.md +++ b/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") ``` . diff --git a/Stable_diff_text2image.py b/Stable_diff_text2image.py index 9a081c4..1e7df6f 100644 --- a/Stable_diff_text2image.py +++ b/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)