Update README.md
Browse files
README.md
CHANGED
@@ -61,7 +61,7 @@ model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloa
|
|
61 |
text = """언어모델은"""
|
62 |
input_ids = tokenizer(text, return_tensors="pt")['input_ids'].to("cuda")
|
63 |
output_ids = model.generate(input_ids, max_new_tokens=100, do_sample=True)
|
64 |
-
print(tokenizer.decode(output_ids[0], skip_special_tokens=
|
65 |
```
|
66 |
<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
|
67 |
|
|
|
61 |
text = """언어모델은"""
|
62 |
input_ids = tokenizer(text, return_tensors="pt")['input_ids'].to("cuda")
|
63 |
output_ids = model.generate(input_ids, max_new_tokens=100, do_sample=True)
|
64 |
+
print(tokenizer.decode(output_ids[0], skip_special_tokens=True))
|
65 |
```
|
66 |
<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
|
67 |
|