[diffusion] fix: webui cannot correctly display generated video using wan2.2 (#18473)
Signed-off-by: Xiaodong Ye <xiaodong.ye@mthreads.com> Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
@@ -111,9 +111,17 @@ def run_sgl_diffusion_webui(server_args: ServerArgs):
|
|||||||
sampling_params_str = "\n".join(
|
sampling_params_str = "\n".join(
|
||||||
[f"{key}: {value}" for key, value in sampling_params_kwargs.items()]
|
[f"{key}: {value}" for key, value in sampling_params_kwargs.items()]
|
||||||
)
|
)
|
||||||
raise ValueError(
|
no_output_msg = f"No output is generated by client, and their sampling params is: {sampling_params_str}"
|
||||||
f"No output is generated by client, and their sampling params is: {sampling_params_str}"
|
|
||||||
)
|
if batch.data_type == DataType.VIDEO:
|
||||||
|
if os.path.exists(save_file_path):
|
||||||
|
logger.warning(no_output_msg)
|
||||||
|
return None, save_file_path
|
||||||
|
else:
|
||||||
|
no_output_msg += f"\nAnd the expected output file was not found at: {save_file_path}"
|
||||||
|
raise ValueError(no_output_msg)
|
||||||
|
else:
|
||||||
|
raise ValueError(no_output_msg)
|
||||||
|
|
||||||
frames = post_process_sample(
|
frames = post_process_sample(
|
||||||
result.output[0],
|
result.output[0],
|
||||||
|
|||||||
Reference in New Issue
Block a user