Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add MPS compatibility and dtype fixes #81

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

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
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