Spaces:
Sleeping
Sleeping
encode plus to encode
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def create_response(input_str,
|
|
47 |
print("num_beams=" + str(num_beams))
|
48 |
seed=int(seed)
|
49 |
print("seed" + str(seed))
|
50 |
-
encoded = tokenizer.
|
51 |
input_ids = encoded["input_ids"]
|
52 |
#attention_mask = encoded["attention_mask"]
|
53 |
|
|
|
47 |
print("num_beams=" + str(num_beams))
|
48 |
seed=int(seed)
|
49 |
print("seed" + str(seed))
|
50 |
+
encoded = tokenizer.encode(input_str + tokenizer.eos_token, return_tensors="pt")
|
51 |
input_ids = encoded["input_ids"]
|
52 |
#attention_mask = encoded["attention_mask"]
|
53 |
|