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

utils.default_device_type(): detect either cuda or mps #718

Open

jimlloyd wants to merge 1 commit into PromtEngineer:main from jimlloyd:auto-mps-detection

jimlloyd commented Jan 26, 2024

Detects if either cuda or mps is available and sets the default device type appropriately.


          utils.default_device_type(): detect either cuda or mps

cd486f2

asegun-cod commented Feb 4, 2024

from utils import default_device_type produces this error on my end.

Traceback (most recent call last):
File "<stdin>", line 1, in <module>
ImportError: cannot import name 'default_device_type' from 'utils' (C:\...\utils.py)

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

Labels

None yet