Update README.md
Browse files
README.md
CHANGED
@@ -58,7 +58,7 @@ This is the first version of Bakpia.
|
|
58 |
- Single-turn QA across various domains.
|
59 |
- Ngoko Javanese not currently supported.
|
60 |
|
61 |
-
##
|
62 |
```
|
63 |
from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer
|
64 |
|
@@ -75,7 +75,6 @@ template = """<|im_start|>system
|
|
75 |
input = template.format(prompt="Kados pundi kulo saged nyinaoni Basa Jawa kanthi sae?"
|
76 |
input = tokenizer([input], return_tensors = "pt").to("cuda")
|
77 |
outputs = model.generate(**input, max_new_tokens = 1024, streamer= TextStreamer(tokenizer), temperature=.5, use_cache=False, do_sample=True)
|
78 |
-
print(tokenizer.batch_decode(outputs)[0])
|
79 |
```
|
80 |
|
81 |
## Acknowledgments
|
|
|
58 |
- Single-turn QA across various domains.
|
59 |
- Ngoko Javanese not currently supported.
|
60 |
|
61 |
+
## Generate with template
|
62 |
```
|
63 |
from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer
|
64 |
|
|
|
75 |
input = template.format(prompt="Kados pundi kulo saged nyinaoni Basa Jawa kanthi sae?"
|
76 |
input = tokenizer([input], return_tensors = "pt").to("cuda")
|
77 |
outputs = model.generate(**input, max_new_tokens = 1024, streamer= TextStreamer(tokenizer), temperature=.5, use_cache=False, do_sample=True)
|
|
|
78 |
```
|
79 |
|
80 |
## Acknowledgments
|