diff --git a/generation_config.json b/generation_config.json new file mode 100644 index 0000000..1a4f0ac --- /dev/null +++ b/generation_config.json @@ -0,0 +1,8 @@ +{ + "bos_token_id": 1, + "eos_token_id": [ + 2, + 6562 + ], + "repetition_penalty": 1.2 +} \ No newline at end of file