nayohan commited on
Commit
6159d8f
1 Parent(s): dd9edfc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -42,7 +42,7 @@ s = "The aerospace industry is a flower in the field of technology and science."
42
  conversation = [{'role': 'system', 'content': SYSTEM_PROMPT},
43
  {'role': 'user', 'content': s}]
44
  inputs = tokenizer.apply_chat_template(conversation,tokenize=True,add_generation_prompt=True,return_tensors='pt').to("cuda")
45
- outputs = model.generate(inputs, early_stopping=True, max_new_tokens=256)
46
  print(tokenizer.decode(outputs[0][len(inputs[0]):]))
47
  ```
48
  ```
 
42
  conversation = [{'role': 'system', 'content': SYSTEM_PROMPT},
43
  {'role': 'user', 'content': s}]
44
  inputs = tokenizer.apply_chat_template(conversation,tokenize=True,add_generation_prompt=True,return_tensors='pt').to("cuda")
45
+ outputs = model.generate(inputs, max_new_tokens=256)
46
  print(tokenizer.decode(outputs[0][len(inputs[0]):]))
47
  ```
48
  ```