AI-Sweden-Models
/

gpt-sw3-6.7b-v2-translator

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

timpal0l commited on Apr 2

Commit

87fd4ce

•

1 Parent(s): 99f6c76

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -39,7 +39,7 @@ class StopOnTokenCriteria(StoppingCriteria):
 pipe = pipeline(
-    "text-generation",
     model="AI-Sweden-Models/gpt-sw3-6.7b-v2-translator",
     device=device
 )

 pipe = pipeline(
+    task="text-generation",
     model="AI-Sweden-Models/gpt-sw3-6.7b-v2-translator",
     device=device
 )