Spaces:
Sleeping
Sleeping
File size: 1,366 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 |
from utils import generate_func, read_prompt_list
import videosys
from videosys import OpenSoraConfig, OpenSoraPipeline
from videosys.models.open_sora import OpenSoraPABConfig
def eval_base(prompt_list):
config = OpenSoraConfig()
pipeline = OpenSoraPipeline(config)
generate_func(pipeline, prompt_list, "./samples/opensora_base", loop=5)
def eval_pab1(prompt_list):
config = OpenSoraConfig(enable_pab=True)
pipeline = OpenSoraPipeline(config)
generate_func(pipeline, prompt_list, "./samples/opensora_pab1", loop=5)
def eval_pab2(prompt_list):
pab_config = OpenSoraPABConfig(spatial_gap=3, temporal_gap=5, cross_gap=7)
config = OpenSoraConfig(enable_pab=True, pab_config=pab_config)
pipeline = OpenSoraPipeline(config)
generate_func(pipeline, prompt_list, "./samples/opensora_pab2", loop=5)
def eval_pab3(prompt_list):
pab_config = OpenSoraPABConfig(spatial_gap=5, temporal_gap=7, cross_gap=9)
config = OpenSoraConfig(enable_pab=True, pab_config=pab_config)
pipeline = OpenSoraPipeline(config)
generate_func(pipeline, prompt_list, "./samples/opensora_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)
|