File size: 1,638 Bytes
07c6a04 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
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)
|