Spaces:
Running
Running
Jaiwincr7
commited on
Commit
·
ccdddd6
1
Parent(s):
13879a6
FINAL DOCKERFIX: Ensuring Dockerfile copies pdf_utils.py instead of the nonexistent main.py.
Browse files- pdf_utils.py +1 -1
pdf_utils.py
CHANGED
|
@@ -21,7 +21,7 @@ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
|
| 21 |
llm = HuggingFacePipeline(
|
| 22 |
pipeline=pipe,
|
| 23 |
model_kwargs={
|
| 24 |
-
"max_new_tokens":
|
| 25 |
"do_sample": True,
|
| 26 |
"temperature": 0.2,
|
| 27 |
"repetition_penalty": 1.05,
|
|
|
|
| 21 |
llm = HuggingFacePipeline(
|
| 22 |
pipeline=pipe,
|
| 23 |
model_kwargs={
|
| 24 |
+
"max_new_tokens": 2000,
|
| 25 |
"do_sample": True,
|
| 26 |
"temperature": 0.2,
|
| 27 |
"repetition_penalty": 1.05,
|