From 8fdabe655dde53b2ff9c98e85626b57e0272fb7b Mon Sep 17 00:00:00 2001 From: arnavsinghvi11 <54859892+arnavsinghvi11@users.noreply.github.com> Date: Wed, 21 Aug 2024 11:33:23 -0700 Subject: [PATCH] Update hf.py --- dsp/modules/hf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dsp/modules/hf.py b/dsp/modules/hf.py index d775d48da..57f57d259 100644 --- a/dsp/modules/hf.py +++ b/dsp/modules/hf.py @@ -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: