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