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

update process tasks function

Browse Source
This commit is contained in:
marksverdhei 2025-08-20 09:51:01 +02:00
parent 63351d419a
commit 61ff48d729
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
wgp.py
View File

@ -4884,6 +4884,7 @@ def generate_preview(model_type, latents):
def process_tasks(state): def process_tasks(state):
from shared.utils.thread_utils import AsyncStream, async_run from shared.utils.thread_utils import AsyncStream, async_run
abort = False
gen = get_gen_info(state) gen = get_gen_info(state)
queue = gen.get("queue", []) queue = gen.get("queue", [])
@ -4919,8 +4920,11 @@ def process_tasks(state):
gen["status"] = "Generating Video" gen["status"] = "Generating Video"
# yield time.time(), time.time() # yield time.time(), time.time()
start_time = time.time()
prompt_no = 0 prompt_no = 0
while len(queue) > 0: while len(queue) > 0:
start_time = time.time()
prompt_no += 1 prompt_no += 1
gen["prompt_no"] = prompt_no gen["prompt_no"] = prompt_no
task = queue[0] task = queue[0]
@ -4959,15 +4963,15 @@ def process_tasks(state):
gen["status"] = data gen["status"] = data
elif cmd == "output": elif cmd == "output":
gen["preview"] = None gen["preview"] = None
# yield time.time() , time.time()
elif cmd == "progress": elif cmd == "progress":
gen["progress_args"] = data gen["progress_args"] = data
# progress(*data) if progress is not None:
progress(*data)
elif cmd == "preview": elif cmd == "preview":
torch.cuda.current_stream().synchronize() torch.cuda.current_stream().synchronize()
preview= None if data== None else generate_preview(params["model_type"], data) preview = None if data== None else generate_preview(params["model_type"], data)
gen["preview"] = preview gen["preview"] = preview
# yield time.time() , gr.Text() # return time.time(), gr.Text()
else: else:
raise Exception(f"unknown command {cmd}") raise Exception(f"unknown command {cmd}")
@ -4985,10 +4989,8 @@ def process_tasks(state):
gen["prompt"] = "" gen["prompt"] = ""
end_time = time.time() end_time = time.time()
if abort: if abort:
# status = f"Video generation was aborted. Total Generation Time: {end_time-start_time:.1f}s"
status = f"Video generation was aborted. Total Generation Time: {format_time(end_time-start_time)}" status = f"Video generation was aborted. Total Generation Time: {format_time(end_time-start_time)}"
else: else:
# status = f"Total Generation Time: {end_time-start_time:.1f}s"
status = f"Total Generation Time: {format_time(end_time-start_time)}" status = f"Total Generation Time: {format_time(end_time-start_time)}"
# Play notification sound when video generation completed successfully # Play notification sound when video generation completed successfully
try: try:
@ -8724,4 +8726,4 @@ if __name__ == "__main__":
else: else:
url = "http://" + server_name url = "http://" + server_name
webbrowser.open(url + ":" + str(server_port), new = 0, autoraise = True) webbrowser.open(url + ":" + str(server_port), new = 0, autoraise = True)
demo.launch(favicon_path="favicon.png", server_name=server_name, server_port=server_port, share=args.share, allowed_paths=list({save_path, image_save_path})) demo.launch(favicon_path="favicon.png", server_name=server_name, server_port=server_port, share=args.share, allowed_paths=list({save_path, image_save_path}))