srowen commited on
Commit
72e3a47
1 Parent(s): 0fe9f64

Update pip example to be consistent with repo

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -38,7 +38,7 @@ To use the model with the `transformers` library on a machine with GPUs, first m
38
  In a Databricks notebook you could run:
39
 
40
  ```python
41
- %pip install accelerate>=0.12.0 transformers[torch]==4.25.1
42
  ```
43
 
44
  The instruction following pipeline can be loaded using the `pipeline` function as shown below. This loads a custom `InstructionTextGenerationPipeline`
 
38
  In a Databricks notebook you could run:
39
 
40
  ```python
41
+ %pip install "accelerate>=0.16.0,<1" "transformers[torch]>=4.28.1,<5" "torch>=1.13.1,<2"
42
  ```
43
 
44
  The instruction following pipeline can be loaded using the `pipeline` function as shown below. This loads a custom `InstructionTextGenerationPipeline`