Skip to content

Conversation

@waijaetee
Copy link

This PR introduces MPS compatibility for Janus-Pro-7B and dtype fixes to improve stability on Apple Silicon.

  • Added app_januspro_mps.py: optimized for Apple MPS, with automatic device selection
  • Fixed dtype mismatch issues in vq_model.py to ensure stability on MPS
  • Updated README.md to document MPS improvements and call for community testing
  • Contribution based on community testing and AI-assisted debugging

Important Notes:

  • This contribution is provided as-is, based on personal testing.
  • I am not a software engineer and will not be maintaining or iterating on this PR.
  • I defer to the project maintainers and community to evaluate, improve, or reject this as they see fit.

- Added app_januspro_mps.py: optimized for Apple MPS, with automatic device selection
- Fixed dtype mismatch issues in vq_model.py to ensure stability on MPS
- Updated README.md to document MPS improvements and call for community testing
- Contribution based on community testing and AI-assisted debugging
@andrusha andrusha mentioned this pull request Jan 30, 2025
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