Simplify example snippet
Browse files
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]
|
| 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 |
|