Commit
·
784b22c
1
Parent(s):
75d52b4
fix requirements
Browse files- README.md +1 -1
- requirements.txt +2 -1
README.md
CHANGED
|
@@ -16,7 +16,7 @@ We recommend <strong>compiled version</strong> of MUDDFormer 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 |
## 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
|