nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-21 11:59:01 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

[feat] fps control

Browse source
This commit is contained in:
zhengzangw 2024-06-14 07:44:35 +00:00
parent 564aec9cba
commit c93e846017
3 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
configs/opensora-v1-2/inference/sample.py
View file

@ -3,6 +3,7 @@ aspect_ratio = "9:16"
num_frames = 51
fps = 24
frame_interval = 1
save_fps = 24
prompt_path = "./assets/texts/t2v_sora.txt"
# == Uncomment the following line to use the prompt below ==

1
opensora/utils/config_utils.py
View file

@ -50,6 +50,7 @@ def parse_args(training=False):
# image/video
parser.add_argument("--num-frames", default=None, type=str, help="number of frames")
parser.add_argument("--fps", default=None, type=int, help="fps")
parser.add_argument("--save-fps", default=None, type=int, help="save fps")
parser.add_argument("--image-size", default=None, type=int, nargs=2, help="image size")
parser.add_argument("--frame-interval", default=None, type=int, help="frame interval")
parser.add_argument("--aspect-ratio", default=None, type=str, help="aspect ratio (h:w)")

2
scripts/inference.py
View file

@ -122,7 +122,7 @@ def main():
# == prepare arguments ==
fps = cfg.fps
save_fps = fps // cfg.get("frame_interval", 1)
save_fps = cfg.get("save_fps", fps // cfg.get("frame_interval", 1))
multi_resolution = cfg.get("multi_resolution", None)
batch_size = cfg.get("batch_size", 1)
num_sample = cfg.get("num_sample", 1)