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

added output path to config

Browse Source
This commit is contained in:
Chris Malone 2025-03-19 12:46:42 +11:00
parent 1406a44a69
commit b727d5174f
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
gradio_server.py
View File

@ -269,6 +269,7 @@ if not Path(server_config_filename).is_file():
"transformer_filename": transformer_choices_t2v[0], "transformer_filename": transformer_choices_t2v[0],
"transformer_filename_i2v": transformer_choices_i2v[1], ######## "transformer_filename_i2v": transformer_choices_i2v[1], ########
"text_encoder_filename" : text_encoder_choices[1], "text_encoder_filename" : text_encoder_choices[1],
"save_path": os.path.join(os.getcwd(), "gradio_outputs"),
"compile" : "", "compile" : "",
"default_ui": "t2v", "default_ui": "t2v",
"boost" : 1, "boost" : 1,
@ -643,6 +644,7 @@ def apply_changes( state,
transformer_t2v_choice, transformer_t2v_choice,
transformer_i2v_choice, transformer_i2v_choice,
text_encoder_choice, text_encoder_choice,
save_path_choice,
attention_choice, attention_choice,
compile_choice, compile_choice,
profile_choice, profile_choice,
@ -660,6 +662,7 @@ def apply_changes( state,
"transformer_filename": transformer_choices_t2v[transformer_t2v_choice], "transformer_filename": transformer_choices_t2v[transformer_t2v_choice],
"transformer_filename_i2v": transformer_choices_i2v[transformer_i2v_choice], ########## "transformer_filename_i2v": transformer_choices_i2v[transformer_i2v_choice], ##########
"text_encoder_filename" : text_encoder_choices[text_encoder_choice], "text_encoder_filename" : text_encoder_choices[text_encoder_choice],
"save_path" : save_path_choice,
"compile" : compile_choice, "compile" : compile_choice,
"profile" : profile_choice, "profile" : profile_choice,
"vae_config" : vae_config_choice, "vae_config" : vae_config_choice,
@ -1088,7 +1091,8 @@ def generate_video(
file_list = [] file_list = []
state["file_list"] = file_list state["file_list"] = file_list
save_path = os.path.join(os.getcwd(), "gradio_outputs") global save_path
save_path = server_config.get("save_path", os.path.join(os.getcwd(), "gradio_outputs"))
os.makedirs(save_path, exist_ok=True) os.makedirs(save_path, exist_ok=True)
video_no = 0 video_no = 0
total_video = repeat_generation * len(prompts) total_video = repeat_generation * len(prompts)
@ -1229,7 +1233,7 @@ def generate_video(
file_name = f"{time_flag}_seed{seed}_{sanitize_file_name(prompt[:50]).strip()}.mp4" file_name = f"{time_flag}_seed{seed}_{sanitize_file_name(prompt[:50]).strip()}.mp4"
else: else:
file_name = f"{time_flag}_seed{seed}_{sanitize_file_name(prompt[:100]).strip()}.mp4" file_name = f"{time_flag}_seed{seed}_{sanitize_file_name(prompt[:100]).strip()}.mp4"
video_path = os.path.join(os.getcwd(), "gradio_outputs", file_name) video_path = os.path.join(save_path, file_name)
cache_video( cache_video(
tensor=sample[None], tensor=sample[None],
save_file=video_path, save_file=video_path,
@ -1675,6 +1679,10 @@ def create_demo():
value= index, value= index,
label="Text Encoder model" label="Text Encoder model"
) )
save_path_choice = gr.Textbox(
label="Output Folder for Generated Videos",
value=server_config.get("save_path", os.path.join(os.getcwd(), "gradio_outputs"))
)
def check(mode): def check(mode):
if not mode in attention_modes_supported: if not mode in attention_modes_supported:
return " (NOT INSTALLED)" return " (NOT INSTALLED)"
@ -2030,6 +2038,7 @@ def create_demo():
transformer_t2v_choice, transformer_t2v_choice,
transformer_i2v_choice, transformer_i2v_choice,
text_encoder_choice, text_encoder_choice,
save_path_choice,
attention_choice, attention_choice,
compile_choice, compile_choice,
profile_choice, profile_choice,