Skip to content
Snippets Groups Projects
Commit 0e9522cc authored by Chen, Eric's avatar Chen, Eric
Browse files

Correction of README

parent f7686863
No related branches found
No related tags found
No related merge requests found
......@@ -37,7 +37,7 @@ python3 models/gpt2_model.py \
--max_time=2
```
For ConditionalGPT2, we have as arguments **--age**, **--sex** and **--anxiety** to generate a persona profile on which we condition the response generation.
For ConditionalGPT2, we have as arguments **--age**, **--sex** and **--topic** to generate a persona profile on which we condition the response generation.
```
python3 models/conditional_gpt2_model.py \
--mode='inference' \
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment