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