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
Files changed (1) hide show
  1. 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": 4096,
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,