[CTRL] warn if generation prompt does not start with a control code

see also https://github.com/salesforce/ctrl/pull/50
2019-10-22 21:27:20 +00:00
parent e16d46843a
commit ef1b8b2ae5
4 changed files with 65 additions and 3 deletions
--- a/README.md
+++ b/README.md
@@ -413,7 +413,7 @@ and from the Salesforce CTRL model:
 python ./examples/run_generation.py \
    --model_type=ctrl \
    --length=20 \
-    --model_name_or_path=gpt2 \
+    --model_name_or_path=ctrl \
    --temperature=0 \
    --repetition_penalty=1.2 \
 ```