Skip to content

Commit

Permalink
Added version in transformer install info
Browse files Browse the repository at this point in the history
  • Loading branch information
Anindyadeep committed Jun 17, 2024
1 parent 27da8c9 commit f24cf8e
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion dsp/modules/tensorrt_llm.py
Original file line number Diff line number Diff line change
Expand Up @@ -108,7 +108,8 @@ def __init__(self, model_name_or_path: str, engine_dir: str, **engine_kwargs: di
from transformers import AutoTokenizer
except ImportError as exc:
raise ModuleNotFoundError(
"You need to install torch and transformers",
"You need to install torch and transformers ",
"pip install transformers==4.38.2",
) from exc

# Configure tokenizer
Expand Down

0 comments on commit f24cf8e

Please sign in to comment.