mirror of
https://github.com/denizsafak/abogen.git
synced 2026-07-18 13:40:27 +02:00
82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
import re
|
|
from typing import List, Tuple
|
|
|
|
from abogen.constants import VOICES_INTERNAL
|
|
|
|
|
|
# Calls parsing and loads the voice to gpu or cpu
|
|
def get_new_voice(pipeline, formula, use_gpu):
|
|
try:
|
|
weighted_voice = parse_voice_formula(pipeline, formula)
|
|
# device = "cuda" if use_gpu else "cpu"
|
|
# Setting the device "cuda" gives "Error occurred: split_with_sizes(): argument 'split_sizes' (position 2)"
|
|
# error when the device is gpu. So disabling this for now.
|
|
device = "cpu"
|
|
return weighted_voice.to(device)
|
|
except Exception as e:
|
|
raise ValueError(f"Failed to create voice: {str(e)}")
|
|
|
|
|
|
def parse_formula_terms(formula: str) -> List[Tuple[str, float]]:
|
|
if not formula or not formula.strip():
|
|
raise ValueError("Empty voice formula")
|
|
|
|
terms: List[Tuple[str, float]] = []
|
|
for segment in formula.split("+"):
|
|
part = segment.strip()
|
|
if not part:
|
|
continue
|
|
if "*" not in part:
|
|
raise ValueError("Each component must be in the form voice*weight")
|
|
voice_name, raw_weight = part.split("*", 1)
|
|
voice_name = voice_name.strip()
|
|
if voice_name not in VOICES_INTERNAL:
|
|
raise ValueError(f"Unknown voice: {voice_name}")
|
|
try:
|
|
weight = float(raw_weight.strip())
|
|
except ValueError as exc:
|
|
raise ValueError(f"Invalid weight for {voice_name}") from exc
|
|
if weight <= 0:
|
|
raise ValueError(f"Weight for {voice_name} must be positive")
|
|
terms.append((voice_name, weight))
|
|
|
|
if not terms:
|
|
raise ValueError("Voice weights must sum to a positive value")
|
|
|
|
return terms
|
|
|
|
|
|
def parse_voice_formula(pipeline, formula):
|
|
terms = parse_formula_terms(formula)
|
|
|
|
total_weight = sum(weight for _, weight in terms)
|
|
if total_weight <= 0:
|
|
raise ValueError("Voice weights must sum to a positive value")
|
|
|
|
weighted_sum = None
|
|
|
|
for voice_name, weight in terms:
|
|
normalized_weight = weight / total_weight if total_weight > 0 else weight
|
|
|
|
voice_tensor = pipeline.load_single_voice(voice_name)
|
|
|
|
if weighted_sum is None:
|
|
weighted_sum = normalized_weight * voice_tensor
|
|
else:
|
|
weighted_sum += normalized_weight * voice_tensor
|
|
|
|
if weighted_sum is None:
|
|
raise ValueError("Voice formula produced no components")
|
|
|
|
return weighted_sum
|
|
|
|
|
|
def calculate_sum_from_formula(formula):
|
|
weights = re.findall(r"\* *([\d.]+)", formula)
|
|
total_sum = sum(float(weight) for weight in weights)
|
|
return total_sum
|
|
|
|
|
|
def extract_voice_ids(formula: str) -> List[str]:
|
|
return [voice for voice, _ in parse_formula_terms(formula)]
|