Commit ·
85607b4
1
Parent(s): 37c6a98
fix requirements
Browse files- README.md +2 -1
- requirements.txt +2 -1
README.md
CHANGED
|
@@ -16,7 +16,8 @@ We recommend <strong>compiled version</strong> of MUDDPythia with *torch.compile
|
|
| 16 |
## Env
|
| 17 |
|
| 18 |
```
|
| 19 |
-
pip install transformers==4.
|
|
|
|
| 20 |
```
|
| 21 |
|
| 22 |
## Generation
|
|
|
|
| 16 |
## Env
|
| 17 |
|
| 18 |
```
|
| 19 |
+
pip install transformers==4.40.2 torch==2.5.1 einops==0.8.0
|
| 20 |
+
|
| 21 |
```
|
| 22 |
|
| 23 |
## Generation
|
requirements.txt
CHANGED
|
@@ -1,2 +1,3 @@
|
|
| 1 |
torch==2.5.1
|
| 2 |
-
transformers==4.
|
|
|
|
|
|
| 1 |
torch==2.5.1
|
| 2 |
+
transformers==4.40.2
|
| 3 |
+
einops==0.8.0
|