Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-07-14 03:30:10 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

logging the generate time

Browse Source
This commit is contained in:
songruizz 2025-06-10 20:59:37 +08:00
parent feba2a62c8
commit 0492d77199
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
generate.py
View File

@ -3,6 +3,7 @@ import argparse
import logging import logging
import os import os
import sys import sys
import time
import warnings import warnings
from tqdm import tqdm from tqdm import tqdm
from datetime import datetime from datetime import datetime
@ -977,6 +978,7 @@ def generate(args):
logging.info( logging.info(
f"Generating {'image' if 't2i' in args.task else 'video'} ...") f"Generating {'image' if 't2i' in args.task else 'video'} ...")
start = time.time()
video = wan_t2v.generate( video = wan_t2v.generate(
args.prompt, args.prompt,
size=SIZE_CONFIGS[args.size], size=SIZE_CONFIGS[args.size],
@ -987,6 +989,7 @@ def generate(args):
guide_scale=args.sample_guide_scale, guide_scale=args.sample_guide_scale,
seed=args.base_seed, seed=args.base_seed,
offload_model=args.offload_model) offload_model=args.offload_model)
logging.info(f"Generate done, elapsed {round(time.time() - start, 2)} s")
elif "i2v" in args.task: elif "i2v" in args.task:
if args.prompt is None: if args.prompt is None:
@ -1066,6 +1069,7 @@ def generate(args):
wan_i2v.model.__class__.cutoff_steps = args.sample_steps*2 - 2 wan_i2v.model.__class__.cutoff_steps = args.sample_steps*2 - 2
logging.info("Generating video ...") logging.info("Generating video ...")
start = time.time()
video = wan_i2v.generate( video = wan_i2v.generate(
args.prompt, args.prompt,
img, img,
@ -1077,6 +1081,7 @@ def generate(args):
guide_scale=args.sample_guide_scale, guide_scale=args.sample_guide_scale,
seed=args.base_seed, seed=args.base_seed,
offload_model=args.offload_model) offload_model=args.offload_model)
logging.info(f"Generate done, elapsed {round(time.time() - start, 2)} s")
elif "flf2v" in args.task: elif "flf2v" in args.task:
if args.prompt is None: if args.prompt is None:
args.prompt = EXAMPLE_PROMPT[args.task]["prompt"] args.prompt = EXAMPLE_PROMPT[args.task]["prompt"]
@ -1124,6 +1129,7 @@ def generate(args):
) )
logging.info("Generating video ...") logging.info("Generating video ...")
start = time.time()
video = wan_flf2v.generate( video = wan_flf2v.generate(
args.prompt, args.prompt,
first_frame, first_frame,
@ -1136,6 +1142,7 @@ def generate(args):
guide_scale=args.sample_guide_scale, guide_scale=args.sample_guide_scale,
seed=args.base_seed, seed=args.base_seed,
offload_model=args.offload_model) offload_model=args.offload_model)
logging.info(f"Generate done, elapsed {round(time.time() - start, 2)} s")
elif "vace" in args.task: elif "vace" in args.task:
if args.prompt is None: if args.prompt is None:
args.prompt = EXAMPLE_PROMPT[args.task]["prompt"] args.prompt = EXAMPLE_PROMPT[args.task]["prompt"]
@ -1178,6 +1185,7 @@ def generate(args):
], args.frame_num, SIZE_CONFIGS[args.size], device) ], args.frame_num, SIZE_CONFIGS[args.size], device)
logging.info(f"Generating video...") logging.info(f"Generating video...")
start = time.time()
video = wan_vace.generate( video = wan_vace.generate(
args.prompt, args.prompt,
src_video, src_video,
@ -1191,6 +1199,7 @@ def generate(args):
guide_scale=args.sample_guide_scale, guide_scale=args.sample_guide_scale,
seed=args.base_seed, seed=args.base_seed,
offload_model=args.offload_model) offload_model=args.offload_model)
logging.info(f"Generate done, elapsed {round(time.time() - start, 2)} s")
else: else:
raise ValueError(f"Unkown task type: {args.task}") raise ValueError(f"Unkown task type: {args.task}")