afrizalha commited on
Commit
e20e52c
1 Parent(s): 76d2b0e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
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
- ## Use
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