Spaces:
Running
Running
Update app.py
Browse files
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(
|