From 72e3a4718631bd5608c3808a9ec85fabce59989f Mon Sep 17 00:00:00 2001 From: Sean Owen Date: Tue, 25 Apr 2023 00:50:54 +0000 Subject: [PATCH] Update pip example to be consistent with repo --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 19295e8..8c4d644 100644 --- a/README.md +++ b/README.md @@ -38,7 +38,7 @@ To use the model with the `transformers` library on a machine with GPUs, first m In a Databricks notebook you could run: ```python -%pip install accelerate>=0.12.0 transformers[torch]==4.25.1 +%pip install "accelerate>=0.16.0,<1" "transformers[torch]>=4.28.1,<5" "torch>=1.13.1,<2" ``` The instruction following pipeline can be loaded using the `pipeline` function as shown below. This loads a custom `InstructionTextGenerationPipeline`