Ethanfel
6bc3fd6443
Pure PyTorch SelVA source for SelvaModelLoader/FeatureExtractor/Sampler nodes. Imports rewritten from selva.* to selva_core.*. mel_converter.py: replaced librosa.filters.mel with pure-numpy implementation to avoid librosa→numba→NumPy version incompatibility in some ComfyUI environments. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
13 lines
361 B
Python
13 lines
361 B
Python
from typing import Optional
|
|
|
|
import torch
|
|
|
|
|
|
def log_normal_sample(x: torch.Tensor,
|
|
generator: Optional[torch.Generator] = None,
|
|
m: float = 0.0,
|
|
s: float = 1.0) -> torch.Tensor:
|
|
bs = x.shape[0]
|
|
s = torch.randn(bs, device=x.device, generator=generator) * s + m
|
|
return torch.sigmoid(s)
|