Update README.md
Browse files
README.md
CHANGED
@@ -28,12 +28,11 @@ def generate(prompt):
|
|
28 |
**data,
|
29 |
generation_config=generation_config
|
30 |
)[0]
|
31 |
-
print(tokenizer.decode(data["input_ids"][0].tolist()))
|
32 |
out = tokenizer.decode(output_ids.tolist())
|
33 |
return out
|
34 |
|
35 |
while 1:
|
36 |
-
generate(input(":> "))
|
37 |
|
38 |
```
|
39 |
# Citation
|
|
|
28 |
**data,
|
29 |
generation_config=generation_config
|
30 |
)[0]
|
|
|
31 |
out = tokenizer.decode(output_ids.tolist())
|
32 |
return out
|
33 |
|
34 |
while 1:
|
35 |
+
print(generate(input(":> ")))
|
36 |
|
37 |
```
|
38 |
# Citation
|