Update pip example to be consistent with repo
Browse files
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.
|
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`
|