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,
|