Update README.md
Browse files
README.md
CHANGED
@@ -55,7 +55,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
55 |
)
|
56 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
57 |
|
58 |
-
inputs = tokenizer("
|
59 |
outputs = model.generate(inputs.input_ids,
|
60 |
max_new_tokens = 1024,
|
61 |
pad_token_id=tokenizer.pad_token_id,
|
|
|
55 |
)
|
56 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
57 |
|
58 |
+
inputs = tokenizer("Bagaimanakah sebuah negara dapat terbentuk?", return_tensors="pt").to("cuda")
|
59 |
outputs = model.generate(inputs.input_ids,
|
60 |
max_new_tokens = 1024,
|
61 |
pad_token_id=tokenizer.pad_token_id,
|