| # Copyright (c) 2025 FoundationVision | |
| # SPDX-License-Identifier: MIT | |
| def get_encode_decode_func(dynamic_scale_schedule): | |
| if dynamic_scale_schedule == 'infinity_star_interact': | |
| from infinity.schedules.infinity_star_interact import video_encode, video_decode, get_visual_rope_embeds, get_scale_pack_info | |
| elif 'infinity_elegant' in dynamic_scale_schedule: | |
| from infinity.schedules.infinity_elegant import video_encode, video_decode, get_visual_rope_embeds, get_scale_pack_info | |
| else: | |
| raise NotImplementedError(f'dynamic_scale_schedule not implemented: {dynamic_scale_schedule}') | |
| return video_encode, video_decode, get_visual_rope_embeds, get_scale_pack_info | |