Update README.md
Browse files
README.md
CHANGED
|
@@ -13,8 +13,6 @@ Pretrained model: [GODEL-v1_1-base-seq2seq](https://huggingface.co/microsoft/GOD
|
|
| 13 |
|
| 14 |
Fine-tuning dataset: [MultiWOZ 2.2](https://github.com/budzianowski/multiwoz/tree/master/data/MultiWOZ_2.2)
|
| 15 |
|
| 16 |
-
(During training, each sample had a maximum of 5 turns of context.)
|
| 17 |
-
|
| 18 |
# How to use:
|
| 19 |
|
| 20 |
```python
|
|
@@ -31,7 +29,7 @@ context = [
|
|
| 31 |
"USER: I'd like to leave on Monday and arrive by 18:00.",
|
| 32 |
]
|
| 33 |
|
| 34 |
-
input_text = " EOS ".join(context) + " => "
|
| 35 |
|
| 36 |
model_inputs = tokenizer(
|
| 37 |
input_text, max_length=512, truncation=True, return_tensors="pt"
|
|
|
|
| 13 |
|
| 14 |
Fine-tuning dataset: [MultiWOZ 2.2](https://github.com/budzianowski/multiwoz/tree/master/data/MultiWOZ_2.2)
|
| 15 |
|
|
|
|
|
|
|
| 16 |
# How to use:
|
| 17 |
|
| 18 |
```python
|
|
|
|
| 29 |
"USER: I'd like to leave on Monday and arrive by 18:00.",
|
| 30 |
]
|
| 31 |
|
| 32 |
+
input_text = " EOS ".join(context[-5:]) + " => "
|
| 33 |
|
| 34 |
model_inputs = tokenizer(
|
| 35 |
input_text, max_length=512, truncation=True, return_tensors="pt"
|