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

Change default device selection #74

Open
chschroeder opened this issue Jan 25, 2025 · 0 comments
Open

Change default device selection #74

chschroeder opened this issue Jan 25, 2025 · 0 comments
Labels
feature request New feature request

Comments

@chschroeder
Copy link
Contributor

Feature description

TransformerBasedClassification and SetFitClassification should select a GPU if available and no device was set (device=None which is the default).

Motivation

Other libraries that small-text uses, such as transformers, sentence-transformers, or setfit check if a GPU device is available and use it by default.

We should adapt the same behavior, this makes it easier for new users and CPU-only usage can still be forced by setting the device parameter.

Additional comments

Thanks to @eisioriginal for reporting

@chschroeder chschroeder added the feature request New feature request label Jan 25, 2025
@chschroeder chschroeder added this to the small-text-2.0.0 milestone Jan 26, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request New feature request
Projects
None yet
Development

No branches or pull requests

1 participant