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 |
|