Skip to content

feat: Add PyTorch 2.6+ compatibility and RTX 5070 support#413

Open
hunterlarcuad wants to merge 1 commit intoTMElyralab:mainfrom
hunterlarcuad:feature/pytorch-2.6-compatibility
Open

feat: Add PyTorch 2.6+ compatibility and RTX 5070 support#413
hunterlarcuad wants to merge 1 commit intoTMElyralab:mainfrom
hunterlarcuad:feature/pytorch-2.6-compatibility

Conversation

@hunterlarcuad
Copy link
Copy Markdown

Summary

  • Add PyTorch 2.6+ compatibility monkey patch for torch.load
  • Add weights_only=False parameter to all torch.load calls
  • Fix relative import in preprocessing.py
  • Add RTX 5070 environment support files
  • Add CLI wrapper for easier inference
  • Optimize download_weights.sh script

Changes

Core Fixes

  • app.py, scripts/inference.py, scripts/realtime_inference.py: Add PyTorch 2.6+ compatibility monkey patch
  • musetalk/models/unet.py, musetalk/utils/face_parsing/, etc.: Add weights_only=False to all torch.load calls
  • musetalk/utils/preprocessing.py: Fix relative import issue

New Files

  • requirements-rtx5070.txt: RTX 5070 specific dependencies
  • run_cli.py: CLI wrapper for easier inference
  • analyze-gemini.md, env-5070.md: Environment analysis documentation
  • reset_mmdet.py: mmdet initialization reset script
  • chumpy-0.70/: chumpy library for compatibility

Optimizations

  • download_weights.sh: Remove unnecessary pip installs, add HuggingFace mirror

Test Plan

  • Test with PyTorch 2.6+
  • Test with RTX 5070 GPU
  • Verify all torch.load calls work correctly
  • Test CLI wrapper functionality

🤖 Generated with Claude Code

- Add PyTorch 2.6+ compatibility monkey patch for torch.load
- Add weights_only=False parameter to all torch.load calls
- Fix relative import in preprocessing.py
- Add RTX 5070 environment support files
- Add CLI wrapper for easier inference
- Optimize download_weights.sh script

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
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.

2 participants