Update README.md
Browse files
README.md
CHANGED
@@ -34,7 +34,11 @@ See the snippet below for usage with Transformers:
|
|
34 |
model_kwargs={"torch_dtype": torch.bfloat16},
|
35 |
device_map="auto"
|
36 |
)
|
37 |
-
>>> pipeline(
|
|
|
|
|
|
|
|
|
38 |
```
|
39 |
```python
|
40 |
>>> "Sommar och sol är det bästa jag vet!
|
|
|
34 |
model_kwargs={"torch_dtype": torch.bfloat16},
|
35 |
device_map="auto"
|
36 |
)
|
37 |
+
>>> pipeline(
|
38 |
+
text_inputs="Sommar och sol är det bästa jag vet",
|
39 |
+
max_length=128,
|
40 |
+
repetition_penalty=1.03)
|
41 |
+
)
|
42 |
```
|
43 |
```python
|
44 |
>>> "Sommar och sol är det bästa jag vet!
|