Fix Space configuration and add missing files - Corregir configuración del Space y agregar archivos faltantes

.gitignore

@@ -0,0 +1,75 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyTorch
+*.pth
+*.pt
+*.ckpt
+*.safetensors
+
+# Audio/Video files
+*.mp4
+*.avi
+*.mov
+*.wav
+*.mp3
+*.flac
+*.aac
+
+# Temporary files
+*.tmp
+*.temp
+temp/
+tmp/
+
+# Logs
+*.log
+logs/
+
+# Environment variables
+.env
+.env.local
+.env.production
+
+# IDE
+.vscode/
+.idea/
+*.swp
+*.swo
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Jupyter
+.ipynb_checkpoints
+
+# Model files (large)
+models/
+checkpoints/
+weights/
+
+# Generated content
+outputs/
+results/
+generated/ 

README.md

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2c7386255dc78815a37e0332b36555cc424e39e367faa5aa306a769133fb7cba
-size 7128
+oid sha256:c812e999cac2fa2550cc44310bfb02fa236c0209b10fe432a4bb875b70aa5237
+size 6706

SPACE_SETUP.md

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e86fcff3e629f22476ad4365f47da2516aceaaccd247ab976de2dbfc34ffacaf
+size 4655

app.py

@@ -3,10 +3,30 @@ import random
 import spaces
 import logging
 import os
+import sys
 from pathlib import Path
 from datetime import datetime
 
+# Configuración para Hugging Face Spaces
+print("🚀 Iniciando VEO3 Free Space...")
+
+# Configurar logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Verificar entorno del Space
+space_id = os.getenv("SPACE_ID")
+if space_id:
+    print(f"✅ Ejecutando en Space: {space_id}")
+
+# Verificar GPU
 import torch
+if torch.cuda.is_available():
+    print(f"✅ GPU disponible: {torch.cuda.get_device_name()}")
+    print(f"✅ Memoria GPU: {torch.cuda.get_device_properties(0).total_memory / 1e9:.1f} GB")
+else:
+    print("⚠️ GPU no disponible - usando CPU")
+
 import numpy as np
 import torchaudio
 from diffusers import AutoencoderKLWan, UniPCMultistepScheduler
@@ -22,9 +42,12 @@ from src.transformer_wan_nag import NagWanTransformer3DModel
 # MMAudio imports
 try:
     import mmaudio
+    print("✅ mmaudio importado correctamente")
 except ImportError:
+    print("📦 Instalando mmaudio...")
     os.system("pip install -e .")
     import mmaudio
+    print("✅ mmaudio instalado e importado")
 
 from mmaudio.eval_utils import (ModelConfig, all_model_cfg, generate as mmaudio_generate, 
                                 load_video, make_video, setup_eval_logging)
@@ -33,6 +56,8 @@ from mmaudio.model.networks import MMAudio, get_my_mmaudio
 from mmaudio.model.sequence_config import SequenceConfig
 from mmaudio.model.utils.features_utils import FeaturesUtils
 
+print("✅ Todas las importaciones completadas")
+
 # NAG Video Settings
 MOD_VALUE = 32
 DEFAULT_DURATION_SECONDS = 4

requirements.txt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0be7d2ec1745f5da5bdd55da84ebe16b5cd9ad5c05f754aa48c07cb948d675be
-size 502
+oid sha256:af9613be75695f3b288927b9219672d10a2bca0f85c067c5dcd22c06cd04fb86
+size 615

setup.py

@@ -0,0 +1,27 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="mmaudio",
+    version="0.1.0",
+    description="MMAudio package for audio generation",
+    packages=find_packages(),
+    install_requires=[
+        "torch>=2.1.0",
+        "torchaudio>=2.1.0",
+        "numpy>=1.21.0",
+        "librosa>=0.8.1",
+        "soundfile",
+        "av",
+        "einops>=0.6",
+        "hydra-core>=1.3.2",
+        "requests",
+        "torchdiffeq",
+        "nitrous-ema",
+        "auraloss",
+        "hydra_colorlog",
+        "tensordict",
+        "colorlog",
+        "open_clip_torch",
+    ],
+    python_requires=">=3.8",
+) 

space_config.py

@@ -0,0 +1,142 @@
+#!/usr/bin/env python3
+"""
+Script de configuración para Hugging Face Space
+Resuelve problemas comunes de configuración y dependencias
+"""
+
+import os
+import sys
+import subprocess
+import logging
+
+# Configurar logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+def check_environment():
+    """Verifica el entorno de ejecución"""
+    logger.info("🔍 Verificando entorno...")
+    
+    # Verificar variables de entorno del Space
+    space_id = os.getenv("SPACE_ID")
+    space_host = os.getenv("SPACE_HOST")
+    
+    if space_id:
+        logger.info(f"✅ Space ID: {space_id}")
+    if space_host:
+        logger.info(f"✅ Space Host: {space_host}")
+    
+    # Verificar GPU
+    try:
+        import torch
+        if torch.cuda.is_available():
+            logger.info(f"✅ GPU disponible: {torch.cuda.get_device_name()}")
+            logger.info(f"✅ Memoria GPU: {torch.cuda.get_device_properties(0).total_memory / 1e9:.1f} GB")
+        else:
+            logger.warning("⚠️ GPU no disponible")
+    except ImportError:
+        logger.error("❌ PyTorch no está instalado")
+
+def install_mmaudio():
+    """Instala el paquete mmaudio local"""
+    logger.info("📦 Instalando mmaudio...")
+    
+    try:
+        # Verificar si mmaudio ya está instalado
+        import mmaudio
+        logger.info("✅ mmaudio ya está instalado")
+        return True
+    except ImportError:
+        pass
+    
+    try:
+        # Instalar mmaudio desde el directorio local
+        result = subprocess.run(
+            ["pip", "install", "-e", "."],
+            capture_output=True,
+            text=True,
+            cwd=os.getcwd()
+        )
+        
+        if result.returncode == 0:
+            logger.info("✅ mmaudio instalado correctamente")
+            return True
+        else:
+            logger.error(f"❌ Error instalando mmaudio: {result.stderr}")
+            return False
+    except Exception as e:
+        logger.error(f"❌ Error en instalación: {e}")
+        return False
+
+def verify_dependencies():
+    """Verifica que todas las dependencias estén instaladas"""
+    logger.info("🔍 Verificando dependencias...")
+    
+    dependencies = [
+        "torch",
+        "torchvision", 
+        "torchaudio",
+        "diffusers",
+        "transformers",
+        "accelerate",
+        "gradio",
+        "numpy",
+        "Pillow",
+        "librosa",
+        "soundfile",
+        "av"
+    ]
+    
+    missing = []
+    for dep in dependencies:
+        try:
+            __import__(dep)
+            logger.info(f"✅ {dep}")
+        except ImportError:
+            logger.error(f"❌ {dep} no está instalado")
+            missing.append(dep)
+    
+    if missing:
+        logger.warning(f"⚠️ Dependencias faltantes: {', '.join(missing)}")
+        return False
+    
+    logger.info("✅ Todas las dependencias están instaladas")
+    return True
+
+def setup_models():
+    """Configura los modelos necesarios"""
+    logger.info("🤖 Configurando modelos...")
+    
+    # Verificar que los directorios de modelos existan
+    model_dirs = ["models", "checkpoints"]
+    for dir_name in model_dirs:
+        if not os.path.exists(dir_name):
+            os.makedirs(dir_name)
+            logger.info(f"📁 Creado directorio: {dir_name}")
+
+def main():
+    """Función principal de configuración"""
+    logger.info("🚀 Iniciando configuración del Space...")
+    
+    # Verificar entorno
+    check_environment()
+    
+    # Instalar mmaudio
+    if not install_mmaudio():
+        logger.error("❌ No se pudo instalar mmaudio")
+        return False
+    
+    # Verificar dependencias
+    if not verify_dependencies():
+        logger.error("❌ Faltan dependencias")
+        return False
+    
+    # Configurar modelos
+    setup_models()
+    
+    logger.info("✅ Configuración completada exitosamente")
+    return True
+
+if __name__ == "__main__":
+    success = main()
+    sys.exit(0 if success else 1)