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`
|