From 9a4d70b216d1cd9c953d2f7b3525841ff1c450d9 Mon Sep 17 00:00:00 2001
From: Friday <3270939387@qq.com>
Date: Fri, 22 Sep 2023 21:33:26 +0800
Subject: [PATCH] sd

---
 Stable_diff_text2image.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Stable_diff_text2image.py b/Stable_diff_text2image.py
index 237cd0f..9a081c4 100644
--- 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