Update convert.py (#6)
Browse files- Update convert.py (a7222ae4346614a4b7ae2b5122e9e7d632970689)
Co-authored-by: Rodri <[email protected]>
- convert.py +1 -1
convert.py
CHANGED
|
@@ -28,7 +28,7 @@ Sample usage:
|
|
| 28 |
|
| 29 |
```
|
| 30 |
python src/transformers/models/mixtral/convert_mixtral_weights_to_hf.py \
|
| 31 |
-
--
|
| 32 |
```
|
| 33 |
|
| 34 |
Thereafter, models can be loaded via:
|
|
|
|
| 28 |
|
| 29 |
```
|
| 30 |
python src/transformers/models/mixtral/convert_mixtral_weights_to_hf.py \
|
| 31 |
+
--input-dir /path/to/downloaded/mixtral/weights --model-size 7B --output-dir /output/path
|
| 32 |
```
|
| 33 |
|
| 34 |
Thereafter, models can be loaded via:
|