Spaces:
Runtime error
Runtime error
Update processing.py
Browse files- processing.py +4 -4
processing.py
CHANGED
@@ -47,10 +47,10 @@ class LazyPipeline:
|
|
47 |
"text-generation",
|
48 |
model=model,
|
49 |
tokenizer=tokenizer,
|
50 |
-
max_new_tokens=
|
51 |
-
temperature=0.
|
52 |
-
top_p = 0.
|
53 |
-
top_k =
|
54 |
repetition_penalty = 1.2,
|
55 |
)
|
56 |
return self.pipeline
|
|
|
47 |
"text-generation",
|
48 |
model=model,
|
49 |
tokenizer=tokenizer,
|
50 |
+
max_new_tokens=800,
|
51 |
+
temperature=0.3,
|
52 |
+
top_p = 0.95,
|
53 |
+
top_k = 5,
|
54 |
repetition_penalty = 1.2,
|
55 |
)
|
56 |
return self.pipeline
|