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.

Sign up for GitHub

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

Jump to bottom

feat: add mps support #22

Open
wants to merge 5 commits into
base: main
Choose a base branch
from
Open

feat: add mps support #22

wants to merge 5 commits into from

Commits on Mar 13, 2024

  1. automatically detect the Python version, if above 3.10 patch collecti… 

    …ons to use collections.abc
    @Fodark
    Fodark committed Mar 13, 2024
    Configuration menu
    Copy the full SHA
    c99e9f1 View commit details
    Browse the repository at this point in the history

  2. use transformers 4.38.1 instead of 4.38.2 as autocast fails on mps on… 

    … latest version
    @Fodark
    Fodark committed Mar 13, 2024
    Configuration menu
    Copy the full SHA
    5db8156 View commit details
    Browse the repository at this point in the history

  3. add an util function to detect platflorm and suitable dtype

    @Fodark
    Fodark committed Mar 13, 2024
    Configuration menu
    Copy the full SHA
    494e622 View commit details
    Browse the repository at this point in the history

  4. load model on detected device and use correct dtype

    @Fodark
    Fodark committed Mar 13, 2024
    Configuration menu
    Copy the full SHA
    40ec649 View commit details
    Browse the repository at this point in the history

  5. modify example in README to use the new device and dtype function

    @Fodark
    Fodark committed Mar 13, 2024
    Configuration menu
    Copy the full SHA
    4eac5b5 View commit details
    Browse the repository at this point in the history