sangmin6600 commited on
Commit
be78d6e
·
verified ·
1 Parent(s): fbe9193

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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=False))
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