|
from utils import generate_func, read_prompt_list |
|
|
|
import videosys |
|
from videosys import OpenSoraPlanConfig, OpenSoraPlanPipeline |
|
from videosys.models.open_sora_plan import OpenSoraPlanPABConfig |
|
|
|
|
|
def eval_base(prompt_list): |
|
config = OpenSoraPlanConfig() |
|
pipeline = OpenSoraPlanPipeline(config) |
|
|
|
generate_func(pipeline, prompt_list, "./samples/opensoraplan_base", loop=5) |
|
|
|
|
|
def eval_pab1(prompt_list): |
|
pab_config = OpenSoraPlanPABConfig( |
|
spatial_gap=2, |
|
temporal_gap=4, |
|
cross_gap=6, |
|
) |
|
config = OpenSoraPlanConfig(enable_pab=True, pab_config=pab_config) |
|
pipeline = OpenSoraPlanPipeline(config) |
|
|
|
generate_func(pipeline, prompt_list, "./samples/opensoraplan_pab1", loop=5) |
|
|
|
|
|
def eval_pab2(prompt_list): |
|
pab_config = OpenSoraPlanPABConfig( |
|
spatial_gap=3, |
|
temporal_gap=5, |
|
cross_gap=7, |
|
) |
|
config = OpenSoraPlanConfig(enable_pab=True, pab_config=pab_config) |
|
pipeline = OpenSoraPlanPipeline(config) |
|
|
|
generate_func(pipeline, prompt_list, "./samples/opensoraplan_pab2", loop=5) |
|
|
|
|
|
def eval_pab3(prompt_list): |
|
pab_config = OpenSoraPlanPABConfig( |
|
spatial_gap=5, |
|
temporal_gap=7, |
|
cross_gap=9, |
|
) |
|
config = OpenSoraPlanConfig(enable_pab=True, pab_config=pab_config) |
|
pipeline = OpenSoraPlanPipeline(config) |
|
|
|
generate_func(pipeline, prompt_list, "./samples/opensoraplan_pab3", loop=5) |
|
|
|
|
|
if __name__ == "__main__": |
|
videosys.initialize(42) |
|
prompt_list = read_prompt_list("vbench/VBench_full_info.json") |
|
eval_base(prompt_list) |
|
eval_pab1(prompt_list) |
|
eval_pab2(prompt_list) |
|
eval_pab3(prompt_list) |
|
|