mrleo1nid/ACE-Step
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

still not working??

Browse Source
This commit is contained in:
Gong Junmin
2025-05-08 19:15:10 +08:00
parent cf227c9c37
commit 3bad5de459
3 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
acestep/pipeline_ace_step.py
+2 -2
View File
@@ -823,6 +823,7 @@ class ACEStepPipeline:
encoder_text_hidden_states_null=None, encoder_text_hidden_states_null=None,
use_erg_lyric=False, use_erg_lyric=False,
use_erg_diffusion=False, use_erg_diffusion=False,
shift=3.0,
retake_random_generators=None, retake_random_generators=None,
retake_variance=0.5, retake_variance=0.5,
add_retake_noise=False, add_retake_noise=False,
@@ -831,7 +832,6 @@ class ACEStepPipeline:
repaint_start=0, repaint_start=0,
repaint_end=0, repaint_end=0,
src_latents=None, src_latents=None,
shift=3.0,
): ):
logger.info( logger.info(
@@ -1308,7 +1308,7 @@ class ACEStepPipeline:
sample=target_latents, sample=target_latents,
return_dict=False, return_dict=False,
omega=omega_scale, omega=omega_scale,
generator=random_generators, generator=random_generators[0],
)[0] )[0]
if is_extend: if is_extend:
acestep/schedulers/scheduling_flow_match_res_multistep.py
+2 -2
View File
@@ -27,7 +27,7 @@ from diffusers.schedulers.scheduling_utils import SchedulerMixin
logger = logging.get_logger(__name__) # pylint: disable=invalid-name logger = logging.get_logger(__name__) # pylint: disable=invalid-name
def get_ancestral_step(sigma_from, sigma_to, eta=1.): def get_ancestral_step(sigma_from, sigma_to, eta=0.0):
"""Calculates the noise level (sigma_down) to step down to and the amount """Calculates the noise level (sigma_down) to step down to and the amount
of noise to add (sigma_up) when doing an ancestral sampling step.""" of noise to add (sigma_up) when doing an ancestral sampling step."""
if not eta: if not eta:
@@ -353,7 +353,7 @@ class FlowMatchResMultiStepScheduler(SchedulerMixin, ConfigMixin):
x = sigma_fn(h) * x + h * (b1 * denoised + b2 * self.old_denoised) x = sigma_fn(h) * x + h * (b1 * denoised + b2 * self.old_denoised)
if self.sigmas[self.step_index + 1] > 0: if self.sigmas[self.step_index + 1] > 0:
init_noise = torch.randn_like(x, device=x.device, generator=generator, dtype=x.dtype) init_noise = torch.randn(x.size(), dtype=x.dtype, layout=x.layout, device=x.device, generator=generator)
x = x + init_noise * s_noise * sigma_up x = x + init_noise * s_noise * sigma_up
self.old_denoised = denoised self.old_denoised = denoised
acestep/ui/components.py
+1
View File
@@ -822,6 +822,7 @@ def create_text2music_ui(
use_erg_tag, use_erg_tag,
use_erg_lyric, use_erg_lyric,
use_erg_diffusion, use_erg_diffusion,
shift,
oss_steps, oss_steps,
guidance_scale_text, guidance_scale_text,
guidance_scale_lyric, guidance_scale_lyric,