Skip to content

🐳 feat: Add complete Docker configuration with English comments#50

Open
Sunwood-ai-labs wants to merge 1 commit intoMeiGen-AI:mainfrom
Sunwood-ai-labsII:feature/docker-compose
Open

🐳 feat: Add complete Docker configuration with English comments#50
Sunwood-ai-labs wants to merge 1 commit intoMeiGen-AI:mainfrom
Sunwood-ai-labsII:feature/docker-compose

Conversation

@Sunwood-ai-labs
Copy link

📄 Description

This PR adds a complete Docker configuration setup for the InfiniteTalk project, with all comments translated to English for better internationalization.

✨ What was Added

  • 🐳 Dockerfile.cu12: Complete CUDA 12.1 optimized Dockerfile

    • Multi-stage build with Python 3.10
    • PyTorch 2.4.1 with CUDA 12.1 support
    • All necessary ML libraries (xformers, flash-attn, etc.)
    • UV package manager integration
    • 13 Japanese comments translated to English
  • 🔧 docker-compose.yml: Complete container orchestration setup

    • Gradio WebUI service configuration
    • GPU resource allocation (NVIDIA runtime)
    • Volume mounts for models and outputs
    • Network and port configuration
    • 9 Japanese comments translated to English
  • 🚫 .dockerignore: Docker build optimization

    • Excludes unnecessary files from build context
    • Prevents cache invalidation from irrelevant changes
  • 📝 .gitignore: Python/ML project optimization

    • Standard Python exclusions
    • ML-specific ignores (weights, outputs, etc.)
    • IDE and OS-specific files

⚙️ Technical Details

  • 🖼️ Base Image: PyTorch 2.4.1 with CUDA 12.1
  • 🎮 GPU Support: NVIDIA Container Runtime
  • 📦 Package Manager: UV for faster Python package installation
  • 💾 Resource Requirements: 128GB RAM, 64GB shared memory
  • 🌐 Ports: WebUI on port 8418

🚀 Usage Examples

docker compose up

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant