afrizalha commited on
Commit
f6a54e6
1 Parent(s): dbcaade

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -81,7 +81,7 @@ template = """<start_of_turn>user
81
 
82
  input = template.format(prompt="Kados pundi kulo saged nyinaoni Basa Jawa kanthi sae?")
83
  input = tokenizer([input], return_tensors = "pt").to("cuda")
84
- outputs = model.generate(**input, max_new_tokens = 1024, streamer= TextStreamer(tokenizer), temperature=.5, use_cache=False, do_sample=True)
85
  ```
86
 
87
  ## Acknowledgments
 
81
 
82
  input = template.format(prompt="Kados pundi kulo saged nyinaoni Basa Jawa kanthi sae?")
83
  input = tokenizer([input], return_tensors = "pt").to("cuda")
84
+ outputs = model.generate(**input, max_new_tokens = 1024, streamer= TextStreamer(tokenizer), temperature=.5, use_cache=True, do_sample=True)
85
  ```
86
 
87
  ## Acknowledgments