SayknowLab commited on
Commit
66c75a2
ยท
verified ยท
1 Parent(s): 9168850

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -49,6 +49,7 @@ def ask_sayknow(query):
49
  prompt = f"{persona_guide}---\n[์ •๋ณด]\n{info}\n[์งˆ๋ฌธ]\n{query}\n[๋‹ต๋ณ€] "
50
 
51
  # ์ด์ „ ๋‹ต๋ณ€ ๋กœ์ง ๊ฐœ์„  (attention_mask ์ถ”๊ฐ€) - ์ด ๋ถ€๋ถ„์€ ์ž˜ ์ž‘๋™ํ•˜๊ณ  ์žˆ์„ ๊ฑฐ์•ผ!
 
52
  encoded_input = tokenizer.encode_plus(
53
  prompt,
54
  return_tensors='pt',
@@ -57,7 +58,7 @@ def ask_sayknow(query):
57
  )
58
  input_ids = encoded_input['input_ids']
59
  attention_mask = encoded_input['attention_mask']
60
-
61
  model.eval()
62
  with torch.no_grad():
63
  gen_ids = model.generate(
 
49
  prompt = f"{persona_guide}---\n[์ •๋ณด]\n{info}\n[์งˆ๋ฌธ]\n{query}\n[๋‹ต๋ณ€] "
50
 
51
  # ์ด์ „ ๋‹ต๋ณ€ ๋กœ์ง ๊ฐœ์„  (attention_mask ์ถ”๊ฐ€) - ์ด ๋ถ€๋ถ„์€ ์ž˜ ์ž‘๋™ํ•˜๊ณ  ์žˆ์„ ๊ฑฐ์•ผ!
52
+ tokenizer.pad_token = tokenizer.eos_token
53
  encoded_input = tokenizer.encode_plus(
54
  prompt,
55
  return_tensors='pt',
 
58
  )
59
  input_ids = encoded_input['input_ids']
60
  attention_mask = encoded_input['attention_mask']
61
+
62
  model.eval()
63
  with torch.no_grad():
64
  gen_ids = model.generate(