Update README.md
Browse files
README.md
CHANGED
|
@@ -28,7 +28,7 @@ Quantized with TensorRT-Model-Optimizer 0.37.0
|
|
| 28 |
Calibrated using the [distilled-roleplay](https://huggingface.co/datasets/agentlans/distilled-roleplay) dataset, tagged in the same ChatML format used to train the Wayfarer and Muse models in the first place. This was accomplished by adding the following code to the start of `hf_ptq.py`:
|
| 29 |
|
| 30 |
```
|
| 31 |
-
|
| 32 |
|
| 33 |
dataset_utils.SUPPORTED_DATASET_CONFIG["distilled-roleplay"] = {
|
| 34 |
"config": {
|
|
|
|
| 28 |
Calibrated using the [distilled-roleplay](https://huggingface.co/datasets/agentlans/distilled-roleplay) dataset, tagged in the same ChatML format used to train the Wayfarer and Muse models in the first place. This was accomplished by adding the following code to the start of `hf_ptq.py`:
|
| 29 |
|
| 30 |
```
|
| 31 |
+
from modelopt.torch.utils import dataset_utils
|
| 32 |
|
| 33 |
dataset_utils.SUPPORTED_DATASET_CONFIG["distilled-roleplay"] = {
|
| 34 |
"config": {
|