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

Save generation parameters in a file

Browse Source
This commit is contained in:
DeepBeepMeep 2025-03-18 23:48:48 +01:00
parent b4afced9ce
commit f3641bf38e
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
gradio_server.py
View File

@ -1237,6 +1237,19 @@ def generate_video(
nrow=1, nrow=1,
normalize=True, normalize=True,
value_range=(-1, 1)) value_range=(-1, 1))
configs = {
'prompt': prompt,
'negative_prompt': negative_prompt,
'resolution': resolution,
'video_length': video_length,
'seed': seed,
'num_inference_steps': num_inference_steps,
}
# Save the configs as json format
with open(video_path.replace('.mp4', '.json'), 'w') as f:
json.dump(configs, f, indent=4)
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)
@ -1306,7 +1319,7 @@ def save_lset(state, lset_name, loras_choices, loras_mult_choices, prompt, save_
full_lset_name_filename = os.path.join(lora_dir, lset_name_filename) full_lset_name_filename = os.path.join(lora_dir, lset_name_filename)
with open(full_lset_name_filename, "w", encoding="utf-8") as writer: with open(full_lset_name_filename, "w", encoding="utf-8") as writer:
writer.write(json.dumps(lset)) writer.write(json.dumps(lset, indent=4))
if lset_name in loras_presets: if lset_name in loras_presets:
gr.Info(f"Lora Preset '{lset_name}' has been updated") gr.Info(f"Lora Preset '{lset_name}' has been updated")
@ -1582,7 +1595,7 @@ def create_demo():
} }
""" """
default_flow_shift = get_default_flow(transformer_filename_i2v if use_image2video else transformer_filename_t2v) default_flow_shift = get_default_flow(transformer_filename_i2v if use_image2video else transformer_filename_t2v)
with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="emerald", neutral_hue="slate", text_size= "md")) as demo: with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="sky", neutral_hue="slate", text_size= "md")) as demo:
state_dict = {} state_dict = {}
if use_image2video: if use_image2video: