mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-10-29 12:40:16 +08:00
add stop token to speed up
This commit is contained in:
1 parent
69d1151e86
commit
57432290ea
1 file changed
+1
@@ -116,6 +116,7 @@ class GPT2Generator:
|
||||
self.batch_size = 1
|
||||
self.stop_token = None
|
||||
self.max_history_tokens = 256
|
||||
self.stop_token = '<|endoftext|>'
|
||||
|
||||
self.model_name = "pytorch-gpt2-xl-aid2-v5"
|
||||
self.model_dir = "models"
|
||||
|
||||
Reference in new issue
Block a user