Update README.md
Browse files
README.md
CHANGED
@@ -39,7 +39,7 @@ class StopOnTokenCriteria(StoppingCriteria):
|
|
39 |
|
40 |
|
41 |
pipe = pipeline(
|
42 |
-
"text-generation",
|
43 |
model="AI-Sweden-Models/gpt-sw3-6.7b-v2-translator",
|
44 |
device=device
|
45 |
)
|
|
|
39 |
|
40 |
|
41 |
pipe = pipeline(
|
42 |
+
task="text-generation",
|
43 |
model="AI-Sweden-Models/gpt-sw3-6.7b-v2-translator",
|
44 |
device=device
|
45 |
)
|