sd

1 year ago · 9a4d70b216
1 changed files with 1 additions and 1 deletions
--- a/Stable_diff_text2image.py
+++ b/Stable_diff_text2image.py
@ -6,7 +6,7 @@ from towhee.operator import PyOperator
 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)'):
+    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