diff --git a/README.md b/README.md index 9d37317..50b41c7 100644 --- a/README.md +++ b/README.md @@ -129,6 +129,28 @@ Terminal Technology Department, Alipay, Ant Group. +### Audio Driven after fix the bug in schedule of inference + +Left: After fixing the bug; Right: Before fixing the bug; +Fix the bug will remove the noise of background. + + + + + + + + +
+ + + +
+ + + + + **(Some demo images above are sourced from image websites. If there is any infringement, we will immediately remove them and apologize.)** ## Installation diff --git a/configs/inference/inference_v2.yaml b/configs/inference/inference_v2.yaml index a501c42..34c0b4f 100644 --- a/configs/inference/inference_v2.yaml +++ b/configs/inference/inference_v2.yaml @@ -25,7 +25,7 @@ unet_additional_kwargs: noise_scheduler_kwargs: beta_start: 0.00085 beta_end: 0.012 - beta_schedule: "linear" + beta_schedule: "scaled_linear" clip_sample: false steps_offset: 1 ### Zero-SNR params @@ -33,4 +33,4 @@ noise_scheduler_kwargs: rescale_betas_zero_snr: True timestep_spacing: "trailing" -sampler: DDIM \ No newline at end of file +sampler: DDIM