Spaces:
Sleeping
Sleeping
arjunanand13
commited on
Commit
•
4777fa9
1
Parent(s):
2477cb2
Update app.py
Browse files
app.py
CHANGED
@@ -49,10 +49,10 @@ generate_text = pipeline(
|
|
49 |
tokenizer=tokenizer,
|
50 |
return_full_text=True,
|
51 |
task='text-generation',
|
52 |
-
stopping_criteria=stopping_criteria,
|
53 |
temperature=0.1,
|
54 |
-
max_new_tokens=
|
55 |
-
repetition_penalty=1.1
|
56 |
)
|
57 |
|
58 |
llm = HuggingFacePipeline(pipeline=generate_text)
|
|
|
49 |
tokenizer=tokenizer,
|
50 |
return_full_text=True,
|
51 |
task='text-generation',
|
52 |
+
# stopping_criteria=stopping_criteria,
|
53 |
temperature=0.1,
|
54 |
+
max_new_tokens=2048,
|
55 |
+
# repetition_penalty=1.1
|
56 |
)
|
57 |
|
58 |
llm = HuggingFacePipeline(pipeline=generate_text)
|