wangzhang commited on
Commit
921daaa
·
1 Parent(s): 95df1b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -13,7 +13,7 @@ MAX_INPUT_TOKEN_LENGTH = 4096
13
  DESCRIPTION = """\
14
  # ChatSDB
15
 
16
- 这是SequioaDB旗下的AI智能大语言模型,训练超过上万条真实数据和7亿参数。
17
  ChatSDB是SequoiaDB旗下的AI智能大语言模型,训练超过上万条真实数据和7亿参数</h3><strong>模型🔗: <a>https://huggingface.co/wangzhang/ChatSDB </a></strong><br><strong>Dataset🔗: <a>https://huggingface.co/datasets/wangzhang/sdb </a></strong><br><strong> API Doc🔗: <a>https://zgg3nzdpswxy4a-80.proxy.runpod.net/docs/ <a> </strong>
18
  """
19
 
@@ -36,9 +36,9 @@ def generate(
36
  chat_history: list[tuple[str, str]],
37
  system_prompt: str,
38
  max_new_tokens: int = 1024,
39
- temperature: float = 0.6,
40
  top_p: float = 0.9,
41
- top_k: int = 50,
42
  repetition_penalty: float = 1.2,
43
  ) -> Iterator[str]:
44
  conversation = []
 
13
  DESCRIPTION = """\
14
  # ChatSDB
15
 
16
+ 这是SequioaDB旗下的AI智能大语言模型,训练超过上万条真实数据和7亿参数。<br>
17
  ChatSDB是SequoiaDB旗下的AI智能大语言模型,训练超过上万条真实数据和7亿参数</h3><strong>模型🔗: <a>https://huggingface.co/wangzhang/ChatSDB </a></strong><br><strong>Dataset🔗: <a>https://huggingface.co/datasets/wangzhang/sdb </a></strong><br><strong> API Doc🔗: <a>https://zgg3nzdpswxy4a-80.proxy.runpod.net/docs/ <a> </strong>
18
  """
19
 
 
36
  chat_history: list[tuple[str, str]],
37
  system_prompt: str,
38
  max_new_tokens: int = 1024,
39
+ temperature: float = 0.2,
40
  top_p: float = 0.9,
41
+ top_k: int = 5,
42
  repetition_penalty: float = 1.2,
43
  ) -> Iterator[str]:
44
  conversation = []