From 605db5b0e2e0a3e7e4090b32e1d0c624641ac476 Mon Sep 17 00:00:00 2001
From: ly119399 <ly119399@alibaba-inc.com>
Date: Sun, 12 Jun 2022 23:39:05 +0800
Subject: [PATCH] update

---
 modelscope/utils/nlp/space/ontology.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/modelscope/utils/nlp/space/ontology.py b/modelscope/utils/nlp/space/ontology.py
index 22e48120..b22d3b3e 100644
--- a/modelscope/utils/nlp/space/ontology.py
+++ b/modelscope/utils/nlp/space/ontology.py
@@ -203,8 +203,9 @@ def get_policy_tokens(prompt_num_for_policy):
 
 # all special tokens definition
 def get_special_tokens(other_tokens):
-    special_tokens = ['<go_r>', '<go_b>', '<go_a>', '<go_d>',
-                      '<eos_u>', '<eos_r>', '<eos_b>', '<eos_a>', '<eos_d>', '<eos_q>',
-                      '<sos_u>', '<sos_r>', '<sos_b>', '<sos_a>', '<sos_d>', '<sos_q>'] \
-                     + db_tokens + other_tokens
+    special_tokens = [
+        '<go_r>', '<go_b>', '<go_a>', '<go_d>', '<eos_u>', '<eos_r>',
+        '<eos_b>', '<eos_a>', '<eos_d>', '<eos_q>', '<sos_u>', '<sos_r>',
+        '<sos_b>', '<sos_a>', '<sos_d>', '<sos_q>'
+    ] + db_tokens + other_tokens
     return special_tokens