xMaulana commited on
Commit
6dcf195
1 Parent(s): e0f8ab3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -151,7 +151,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
151
 
152
  inputs = tokenizer("Bagaimanakah sebuah negara dapat terbentuk?", return_tensors="pt").to("cuda")
153
  outputs = model.generate(inputs.input_ids,
154
- max_new_tokens = 1024,
155
  pad_token_id=tokenizer.pad_token_id,
156
  eos_token_id=tokenizer.eos_token_id,
157
  temperature=0.7,
 
151
 
152
  inputs = tokenizer("Bagaimanakah sebuah negara dapat terbentuk?", return_tensors="pt").to("cuda")
153
  outputs = model.generate(inputs.input_ids,
154
+ max_new_tokens = 2048,
155
  pad_token_id=tokenizer.pad_token_id,
156
  eos_token_id=tokenizer.eos_token_id,
157
  temperature=0.7,