Hawk Lee
eff537d9dd
- Add AIIA_ASR node with FunASR (paraformer-zh + SenseVoiceSmall) - Add AIIA_Podcast_Splitter for per-speaker text splitting - Add AIIA_Podcast_Stitcher with 3-tier fuzzy alignment (exact, Levenshtein, fallback) - Register 5 previously missing modules in __init__.py - Update README with installation guide and node documentation
30 lines
778 B
Python
30 lines
778 B
Python
import torch
|
|
|
|
tensor_interpolation = None
|
|
|
|
|
|
def get_tensor_interpolation_method():
|
|
return tensor_interpolation
|
|
|
|
|
|
def set_tensor_interpolation_method(is_slerp):
|
|
global tensor_interpolation
|
|
tensor_interpolation = slerp if is_slerp else linear
|
|
|
|
|
|
def linear(v1, v2, t):
|
|
return (1.0 - t) * v1 + t * v2
|
|
|
|
|
|
def slerp(
|
|
v0: torch.Tensor, v1: torch.Tensor, t: float, DOT_THRESHOLD: float = 0.9995
|
|
) -> torch.Tensor:
|
|
u0 = v0 / v0.norm()
|
|
u1 = v1 / v1.norm()
|
|
dot = (u0 * u1).sum()
|
|
if dot.abs() > DOT_THRESHOLD:
|
|
# logger.info(f'warning: v0 and v1 close to parallel, using linear interpolation instead.')
|
|
return (1.0 - t) * v0 + t * v1
|
|
omega = dot.acos()
|
|
return (((1.0 - t) * omega).sin() * v0 + (t * omega).sin() * v1) / omega.sin()
|