seawolf2357 commited on
Commit
835acd8
ยท
verified ยท
1 Parent(s): 023708d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=25000, 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) # ์‚ฌ์šฉ ๊ฐ€๋Šฅํ•œ ์ตœ๋Œ€ ํ† ํฐ ์ˆ˜ ์กฐ์ •
 
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) # ์‚ฌ์šฉ ๊ฐ€๋Šฅํ•œ ์ตœ๋Œ€ ํ† ํฐ ์ˆ˜ ์กฐ์ •