DataSnake commited on
Commit
3ae6ca3
·
verified ·
1 Parent(s): 7cf2b0a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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
- import modelopt.torch.utils import dataset_utils
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": {