mirror of
https://github.com/Wan-Video/Wan2.1.git
synced 2025-11-04 14:16:57 +00:00
add defaults for missing values in queue.zip
This commit is contained in:
parent
7b2cafb7f9
commit
f23bbf83be
16
wgp.py
16
wgp.py
@ -5739,6 +5739,22 @@ def process_tasks(state):
|
|||||||
send_cmd = com_stream.output_queue.push
|
send_cmd = com_stream.output_queue.push
|
||||||
def generate_video_error_handler():
|
def generate_video_error_handler():
|
||||||
try:
|
try:
|
||||||
|
import inspect
|
||||||
|
model_type = params.get('model_type')
|
||||||
|
known_defaults = {
|
||||||
|
'image_refs_relative_size': 50,
|
||||||
|
}
|
||||||
|
|
||||||
|
for arg_name, default_value in known_defaults.items():
|
||||||
|
if arg_name not in params:
|
||||||
|
print(f"Warning: Missing argument '{arg_name}' in loaded task. Applying default value: {default_value}")
|
||||||
|
params[arg_name] = default_value
|
||||||
|
if model_type:
|
||||||
|
default_settings = get_default_settings(model_type)
|
||||||
|
expected_args = inspect.signature(generate_video).parameters.keys()
|
||||||
|
for arg_name in expected_args:
|
||||||
|
if arg_name not in params and arg_name in default_settings:
|
||||||
|
params[arg_name] = default_settings[arg_name]
|
||||||
generate_video(task, send_cmd, **params)
|
generate_video(task, send_cmd, **params)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
tb = traceback.format_exc().split('\n')[:-1]
|
tb = traceback.format_exc().split('\n')[:-1]
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user