Ethanfel/ComfyUI-SelVA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add fps INT output to PrismAudioFeatureExtractor

Browse Source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Ethanfel
2026-03-28 11:05:03 +01:00
parent 5b62be0447
commit d0c9a72782
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/feature_extractor.py
+4 -3
View File
@@ -105,8 +105,8 @@ class PrismAudioFeatureExtractor:
}, },
} }
RETURN_TYPES = ("PRISMAUDIO_FEATURES",) RETURN_TYPES = ("PRISMAUDIO_FEATURES", "INT")
RETURN_NAMES = ("features",) RETURN_NAMES = ("features", "fps")
FUNCTION = "extract_features" FUNCTION = "extract_features"
CATEGORY = PRISMAUDIO_CATEGORY CATEGORY = PRISMAUDIO_CATEGORY
@@ -134,7 +134,8 @@ class PrismAudioFeatureExtractor:
if os.path.exists(cached_path): if os.path.exists(cached_path):
print(f"[PrismAudio] Using cached features: {cached_path}") print(f"[PrismAudio] Using cached features: {cached_path}")
loader = PrismAudioFeatureLoader() loader = PrismAudioFeatureLoader()
return loader.load_features(cached_path) features, = loader.load_features(cached_path)
return (features, round(fps))
# Save frames to temp file (lossless .npy, no codec roundtrip) # Save frames to temp file (lossless .npy, no codec roundtrip)
import time import time