NeMo
PyTorch
nemotron
srvm commited on
Commit
dd5c82d
·
1 Parent(s): 45afb20

Simplify example snippet

Browse files
Files changed (1) hide show
  1. README.md +2 -5
README.md CHANGED
@@ -42,13 +42,10 @@ prompt = 'Complete the paragraph: our solar system is'
42
  inputs = tokenizer.encode(prompt, return_tensors='pt').to(model.device)
43
 
44
  # Generate the output
45
- outputs = model.generate(inputs,
46
- max_length=20,
47
- num_return_sequences=1,
48
- pad_token_id=tokenizer.eos_token_id)
49
 
50
  # Decode and print the output
51
- output_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
52
  print(output_text)
53
  ```
54
 
 
42
  inputs = tokenizer.encode(prompt, return_tensors='pt').to(model.device)
43
 
44
  # Generate the output
45
+ outputs = model.generate(inputs, max_length=20)
 
 
 
46
 
47
  # Decode and print the output
48
+ output_text = tokenizer.decode(outputs[0])
49
  print(output_text)
50
  ```
51