Skip to content

Commit c24a86b

Browse files
authored
change gs video path to run_name/videos (#241)
1 parent 043f826 commit c24a86b

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/maxdiffusion/generate_wan.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def inference_generate_video(config, pipeline, filename_prefix=""):
8787
video_path = f"{filename_prefix}wan_output_{config.seed}_{i}.mp4"
8888
export_to_video(videos[i], video_path, fps=config.fps)
8989
if config.output_dir.startswith("gs://"):
90-
upload_video_to_gcs(config.output_dir, video_path)
90+
upload_video_to_gcs(os.path.join(config.output_dir, config.run_name), video_path)
9191
# Delete local files to avoid storing too manys videos
9292
delete_file(f"./{video_path}")
9393
return
@@ -128,7 +128,7 @@ def run(config, pipeline=None, filename_prefix=""):
128128
export_to_video(videos[i], video_path, fps=config.fps)
129129
saved_video_path.append(video_path)
130130
if config.output_dir.startswith("gs://"):
131-
upload_video_to_gcs(config.output_dir, video_path)
131+
upload_video_to_gcs(os.path.join(config.output_dir, config.run_name), video_path)
132132

133133
s0 = time.perf_counter()
134134
videos = pipeline(

0 commit comments

Comments
 (0)