Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-11-05 06:29:14 +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() end_time = time.time()
abort = True abort = True
state["prompt"] = "" 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: else:
sample = samples.cpu() sample = samples.cpu()
# video = rearrange(sample.cpu().numpy(), "c t h w -> t h w c") # 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) print(f"New video saved to Path: "+video_path)
file_list.append(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 seed += 1
repeat_no += 1 repeat_no += 1