Skip to content

Commit c303312

Browse files
committed
simplify logic
1 parent f8ee74d commit c303312

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

backend/python/diffusers/backend_diffusers.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -164,11 +164,6 @@ def LoadModel(self, request, context):
164164

165165
fromSingleFile = request.Model.startswith("http") or request.Model.startswith("/") or local
166166

167-
168-
# If PipelineType is not defined, defaults to AutoPipelineForText2Image
169-
if request.PipelineType == "":
170-
request.PipelineType = "AutoPipelineForText2Image"
171-
172167
## img2img
173168
if (request.PipelineType == "StableDiffusionImg2ImgPipeline") or (request.IMG2IMG and request.PipelineType == ""):
174169
if fromSingleFile:
@@ -185,7 +180,7 @@ def LoadModel(self, request, context):
185180
torch_dtype=torchType,
186181
guidance_scale=cfg_scale)
187182
## text2img
188-
elif request.PipelineType == "AutoPipelineForText2Image":
183+
elif request.PipelineType == "AutoPipelineForText2Image" or request.PipelineType == "":
189184
self.pipe = AutoPipelineForText2Image.from_pretrained(request.Model,
190185
torch_dtype=torchType,
191186
use_safetensors=SAFETENSORS,

0 commit comments

Comments
 (0)