Skip to content

Commit

Permalink
Updated warning message
Browse files Browse the repository at this point in the history
  • Loading branch information
hh-space-invader committed Dec 19, 2024
1 parent 7cae308 commit 61188fb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions fastembed/common/onnx_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,9 +54,9 @@ def _load_onnx_model(

if cuda and providers is not None:
warnings.warn(
"You have specified both `cuda=True` and `providers`. This creates a conflict because `providers` takes precedence. "
"Avoid specifying `providers` and use `cuda=True` at the same time.",
f"`cuda` and `providers` are mutually exclusive parameters, cuda: {cuda}, providers: {providers}",
category=UserWarning,
stacklevel=6,
)

if providers is not None:
Expand Down

0 comments on commit 61188fb

Please sign in to comment.