(original_config)
| 306 | |
| 307 | |
| 308 | def create_diffusers_schedular(original_config): |
| 309 | schedular = DDIMScheduler( |
| 310 | num_train_timesteps=original_config.model.params.timesteps, |
| 311 | beta_start=original_config.model.params.linear_start, |
| 312 | beta_end=original_config.model.params.linear_end, |
| 313 | beta_schedule="scaled_linear", |
| 314 | ) |
| 315 | return schedular |
| 316 | |
| 317 | |
| 318 | def create_ldm_bert_config(original_config): |
nothing calls this directly
no outgoing calls
no test coverage detected