diodiogod
b21d0212fb
🔧 ENGINES SEPARATION: - Move chatterbox/ → engines/chatterbox/ (TTS/VC engine + models) - Move chatterbox/f5tts/ → engines/f5tts/ (F5-TTS engine) - Move adapters/ → engines/adapters/ (engine adapters for multilingual system) 📁 NODES ORGANIZATION: - Create nodes/chatterbox/ for ChatterBox nodes (TTS, VC, SRT) - Create nodes/f5tts/ for F5-TTS nodes (TTS, SRT, Edit, Options) - Create nodes/audio/ for audio processing nodes (analyzer, recorder) - Create nodes/base/ for base classes - Rename node files for clarity (tts_node.py → chatterbox_tts_node.py) 🗂️ UTILITIES RESTRUCTURE: - Replace core/ with organized utils/ structure - utils/audio/ - Audio processing, analysis, caching - utils/text/ - Text processing, character parsing, chunking - utils/voice/ - Voice discovery, multilingual engine - utils/timing/ - SRT timing logic (from chatterbox_srt/) - utils/models/ - Model management (ChatterBox + F5-TTS) - utils/system/ - System utilities, import management 🔄 IMPORT PATH UPDATES: - Update all import paths to use new organized structure - Fix lazy imports in nodes for runtime compatibility - Restore missing F5-TTS edit engine from git history - Fix path resolution for voice discovery and character aliases 🎯 KEY IMPROVEMENTS: - Engine-centric organization with clear separation - Purpose-driven folder grouping for better navigation - Self-documenting file names indicating ownership - Scalable structure for future engine additions - Maintains backward compatibility for existing workflows All 10 nodes loading successfully with complete functionality preserved.
31 lines
625 B
Python
31 lines
625 B
Python
from .s3tokenizer import (
|
|
S3_SR,
|
|
S3_HOP,
|
|
S3_TOKEN_HOP,
|
|
S3_TOKEN_RATE,
|
|
SPEECH_VOCAB_SIZE,
|
|
S3Tokenizer,
|
|
)
|
|
|
|
|
|
SOS = SPEECH_VOCAB_SIZE
|
|
EOS = SPEECH_VOCAB_SIZE + 1
|
|
|
|
|
|
|
|
def drop_invalid_tokens(x):
|
|
"""Drop SoS and EoS"""
|
|
assert len(x.shape) == 1 or (len(x.shape) == 2 and x.shape[0] == 1), "only batch size of one allowed for now"
|
|
if SOS in x:
|
|
s = (x == SOS).nonzero(as_tuple=True)[0].squeeze(0) + 1
|
|
else:
|
|
s = 0
|
|
|
|
if EOS in x:
|
|
e = (x == EOS).nonzero(as_tuple=True)[0].squeeze(0)
|
|
else:
|
|
e = None
|
|
|
|
x = x[s: e]
|
|
return x
|