Skip to content

Commit

Permalink
Update hf.py
Browse files Browse the repository at this point in the history
  • Loading branch information
arnavsinghvi11 authored Aug 21, 2024
1 parent bdc9b98 commit 8fdabe6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dsp/modules/hf.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ def __init__(
from transformers import AutoConfig, AutoModelForCausalLM, AutoModelForSeq2SeqLM, AutoTokenizer
except ImportError as exc:
raise ModuleNotFoundError(
"You need to install Hugging Face transformers (with torch dependencies) library to use HF models.",
"You need to install Hugging Face transformers (with torch dependencies - pip install transformers[torch]) library to use HF models.",
) from exc
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
try:
Expand Down

0 comments on commit 8fdabe6

Please sign in to comment.