Update app.py
Browse files
app.py
CHANGED
@@ -83,7 +83,7 @@ def format_prompt(message, history):
|
|
83 |
prompt += f"[INST] {message} [/INST]"
|
84 |
return prompt
|
85 |
|
86 |
-
def generate(prompt, history=[], temperature=0.1, max_new_tokens=
|
87 |
input_tokens = len(tokenizer.encode(prompt))
|
88 |
available_tokens = 32768 - input_tokens # ์ต๋ ํ์ฉ ํ ํฐ ์์์ ์
๋ ฅ ํ ํฐ ์๋ฅผ ๋บ ๊ฐ
|
89 |
max_new_tokens = min(max_new_tokens, available_tokens) # ์ฌ์ฉ ๊ฐ๋ฅํ ์ต๋ ํ ํฐ ์ ์กฐ์
|
|
|
83 |
prompt += f"[INST] {message} [/INST]"
|
84 |
return prompt
|
85 |
|
86 |
+
def generate(prompt, history=[], temperature=0.1, max_new_tokens=23200, top_p=0.95, repetition_penalty=1.0):
|
87 |
input_tokens = len(tokenizer.encode(prompt))
|
88 |
available_tokens = 32768 - input_tokens # ์ต๋ ํ์ฉ ํ ํฐ ์์์ ์
๋ ฅ ํ ํฐ ์๋ฅผ ๋บ ๊ฐ
|
89 |
max_new_tokens = min(max_new_tokens, available_tokens) # ์ฌ์ฉ ๊ฐ๋ฅํ ์ต๋ ํ ํฐ ์ ์กฐ์
|