dependency_check.py hinzugefügt
Some checks are pending
precommit / Run precommit (push) Waiting to run
Some checks are pending
precommit / Run precommit (push) Waiting to run
This commit is contained in:
parent
f23d1be027
commit
34e58b8055
67
dependency_check.py
Normal file
67
dependency_check.py
Normal file
|
|
@ -0,0 +1,67 @@
|
||||||
|
#!/usr/bin/env python3
|
||||||
|
"""
|
||||||
|
Check if all Kyutai TTS dependencies are properly installed
|
||||||
|
"""
|
||||||
|
|
||||||
|
import sys
|
||||||
|
|
||||||
|
def check_dependencies():
|
||||||
|
print("🔍 Checking Kyutai TTS Dependencies")
|
||||||
|
print("=" * 40)
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
"torch",
|
||||||
|
"numpy",
|
||||||
|
"einops",
|
||||||
|
"transformers",
|
||||||
|
"accelerate",
|
||||||
|
"soundfile",
|
||||||
|
"librosa",
|
||||||
|
"huggingface_hub",
|
||||||
|
"moshi",
|
||||||
|
"sphn"
|
||||||
|
]
|
||||||
|
|
||||||
|
missing = []
|
||||||
|
installed = []
|
||||||
|
|
||||||
|
for dep in dependencies:
|
||||||
|
try:
|
||||||
|
__import__(dep)
|
||||||
|
installed.append(dep)
|
||||||
|
print(f"✓ {dep}")
|
||||||
|
except ImportError as e:
|
||||||
|
missing.append((dep, str(e)))
|
||||||
|
print(f"✗ {dep}: {e}")
|
||||||
|
|
||||||
|
print(f"\n📊 Summary:")
|
||||||
|
print(f"✓ Installed: {len(installed)}")
|
||||||
|
print(f"✗ Missing: {len(missing)}")
|
||||||
|
|
||||||
|
if missing:
|
||||||
|
print(f"\n🔧 To fix missing dependencies:")
|
||||||
|
for dep, error in missing:
|
||||||
|
print(f"pip install {dep}")
|
||||||
|
|
||||||
|
print(f"\n🧪 Testing Kyutai TTS imports:")
|
||||||
|
try:
|
||||||
|
from moshi.models.loaders import CheckpointInfo
|
||||||
|
print("✓ CheckpointInfo import successful")
|
||||||
|
except Exception as e:
|
||||||
|
print(f"✗ CheckpointInfo import failed: {e}")
|
||||||
|
|
||||||
|
try:
|
||||||
|
from moshi.models.tts import DEFAULT_DSM_TTS_REPO, DEFAULT_DSM_TTS_VOICE_REPO, TTSModel
|
||||||
|
print("✓ TTSModel imports successful")
|
||||||
|
except Exception as e:
|
||||||
|
print(f"✗ TTSModel imports failed: {e}")
|
||||||
|
|
||||||
|
return len(missing) == 0
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
success = check_dependencies()
|
||||||
|
if success:
|
||||||
|
print("\n🎉 All dependencies are installed correctly!")
|
||||||
|
else:
|
||||||
|
print("\n❌ Some dependencies are missing. Please install them first.")
|
||||||
|
sys.exit(1)
|
||||||
Loading…
Reference in New Issue
Block a user