[Diffusion] fix serving image_edit get input image bug (#18109)
This commit is contained in:
@@ -333,6 +333,9 @@ class DiffusersExecutionStage(PipelineStage):
|
|||||||
if not batch.image_path:
|
if not batch.image_path:
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
if isinstance(batch.image_path, list):
|
||||||
|
batch.image_path = batch.image_path[0]
|
||||||
|
|
||||||
try:
|
try:
|
||||||
if batch.image_path.startswith(("http://", "https://")):
|
if batch.image_path.startswith(("http://", "https://")):
|
||||||
response = requests.get(batch.image_path, timeout=30)
|
response = requests.get(batch.image_path, timeout=30)
|
||||||
|
|||||||
Reference in New Issue
Block a user