Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-11-04 14:16:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

re-removed yield statements that prevented it running

Browse Source
This commit is contained in:
Chris Malone 2025-03-26 18:05:07 +11:00
parent ffe656d897
commit 85d7320837
1 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
gradio_server.py
View File

@ -1431,7 +1431,7 @@ def generate_video(
end_time = time.time()
abort = True
state["prompt"] = ""
yield f"Video generation was aborted. Total Generation Time: {end_time-start_time:.1f}s"
print(f"Video generation was aborted. Total Generation Time: {end_time-start_time:.1f}s")
else:
sample = samples.cpu()
# video = rearrange(sample.cpu().numpy(), "c t h w -> t h w c")
@ -1465,12 +1465,6 @@ def generate_video(
print(f"New video saved to Path: "+video_path)
file_list.append(video_path)
if video_no < total_video:
yield status
else:
end_time = time.time()
state["prompt"] = ""
yield f"Total Generation Time: {end_time-start_time:.1f}s"
seed += 1
repeat_no += 1