(
steps: int = 40,
sampler: str = "EulerEDMSampler",
discretization: str = "EDMShiftDiscretization",
guider: str = "VanillaCFG",
cfg_scale: float = 7.5,
n_context_frames: int = 5
)
| 95 | |
| 96 | |
| 97 | def init_sampling( |
| 98 | steps: int = 40, |
| 99 | sampler: str = "EulerEDMSampler", |
| 100 | discretization: str = "EDMShiftDiscretization", |
| 101 | guider: str = "VanillaCFG", |
| 102 | cfg_scale: float = 7.5, |
| 103 | n_context_frames: int = 5 |
| 104 | ): |
| 105 | discretization_config = get_discretization(discretization) |
| 106 | guider_config = get_guider(guider, cfg_scale) |
| 107 | sampler = get_sampler(sampler, steps, discretization_config, guider_config, n_context_frames) |
| 108 | return sampler |
| 109 | |
| 110 | |
| 111 | def get_discretization(discretization): |
no test coverage detected