(json_data, retake_variance, retake_seeds)
| 324 | retake_outputs, retake_input_params_json = create_output_ui("Retake") |
| 325 | |
| 326 | def retake_process_func(json_data, retake_variance, retake_seeds): |
| 327 | return text2music_process_func( |
| 328 | json_data["format"], |
| 329 | json_data["audio_duration"], |
| 330 | json_data["prompt"], |
| 331 | json_data["lyrics"], |
| 332 | json_data["infer_step"], |
| 333 | json_data["guidance_scale"], |
| 334 | json_data["scheduler_type"], |
| 335 | json_data["cfg_type"], |
| 336 | json_data["omega_scale"], |
| 337 | ", ".join(map(str, json_data["actual_seeds"])), |
| 338 | json_data["guidance_interval"], |
| 339 | json_data["guidance_interval_decay"], |
| 340 | json_data["min_guidance_scale"], |
| 341 | json_data["use_erg_tag"], |
| 342 | json_data["use_erg_lyric"], |
| 343 | json_data["use_erg_diffusion"], |
| 344 | ", ".join(map(str, json_data["oss_steps"])), |
| 345 | ( |
| 346 | json_data["guidance_scale_text"] |
| 347 | if "guidance_scale_text" in json_data |
| 348 | else 0.0 |
| 349 | ), |
| 350 | ( |
| 351 | json_data["guidance_scale_lyric"] |
| 352 | if "guidance_scale_lyric" in json_data |
| 353 | else 0.0 |
| 354 | ), |
| 355 | retake_seeds=retake_seeds, |
| 356 | retake_variance=retake_variance, |
| 357 | task="retake", |
| 358 | lora_name_or_path="none" if "lora_name_or_path" not in json_data else json_data["lora_name_or_path"], |
| 359 | lora_weight=1 if "lora_weight" not in json_data else json_data["lora_weight"] |
| 360 | ) |
| 361 | |
| 362 | retake_bnt.click( |
| 363 | fn=retake_process_func, |
nothing calls this directly
no outgoing calls
no test coverage detected