logo

text2image
/
stable-diffusion
repo-copy-icon

copied

Browse Source

sd

main
Friday 2 years ago
parent
fb2f1c3194
commit
9a4d70b216
1 changed files with 1 additions and 1 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      Stable_diff_text2image.py

2
Stable_diff_text2image.py
View File

@ -6,7 +6,7 @@ from towhee.operator import PyOperator
log = logging.getLogger(PyOperator) log = logging.getLogger(PyOperator)
class StableDiffusion(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._model_id=model_id
self.pipe = pipe self.pipe = pipe

Write Preview
Loading…
Cancel
Save