mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-03-22 01:23:43 +08:00
Merge ddd0cd7521 into ed7c2c6579
This commit is contained in:
commit
00bb6520fe
@ -298,11 +298,11 @@ class ACE15TEModel(torch.nn.Module):
|
|||||||
self.qwen3_06b.set_clip_options({"layer": [0]})
|
self.qwen3_06b.set_clip_options({"layer": [0]})
|
||||||
lyrics_embeds, _, extra_l = self.qwen3_06b.encode_token_weights(token_weight_pairs_lyrics)
|
lyrics_embeds, _, extra_l = self.qwen3_06b.encode_token_weights(token_weight_pairs_lyrics)
|
||||||
|
|
||||||
out = {"conditioning_lyrics": lyrics_embeds[:, 0]}
|
out = {"conditioning_lyrics": lyrics_embeds}
|
||||||
|
|
||||||
lm_metadata = token_weight_pairs["lm_metadata"]
|
lm_metadata = token_weight_pairs["lm_metadata"]
|
||||||
if lm_metadata["generate_audio_codes"]:
|
if lm_metadata["generate_audio_codes"]:
|
||||||
audio_codes = generate_audio_codes(getattr(self, self.lm_model, self.qwen3_06b), token_weight_pairs["lm_prompt"], token_weight_pairs["lm_prompt_negative"], min_tokens=lm_metadata["min_tokens"], max_tokens=lm_metadata["min_tokens"], seed=lm_metadata["seed"], cfg_scale=lm_metadata["cfg_scale"], temperature=lm_metadata["temperature"], top_p=lm_metadata["top_p"], top_k=lm_metadata["top_k"], min_p=lm_metadata["min_p"])
|
audio_codes = generate_audio_codes(getattr(self, self.lm_model, self.qwen3_06b), token_weight_pairs["lm_prompt"], token_weight_pairs["lm_prompt_negative"], min_tokens=lm_metadata["min_tokens"], max_tokens=lm_metadata["max_tokens"], seed=lm_metadata["seed"], cfg_scale=lm_metadata["cfg_scale"], temperature=lm_metadata["temperature"], top_p=lm_metadata["top_p"], top_k=lm_metadata["top_k"], min_p=lm_metadata["min_p"])
|
||||||
out["audio_codes"] = [audio_codes]
|
out["audio_codes"] = [audio_codes]
|
||||||
|
|
||||||
return base_out, None, out
|
return base_out, None, out
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user