green commited on
Commit
c952a7d
1 Parent(s): bdb86eb

Update digestor.py

Browse files
Files changed (1) hide show
  1. digestor.py +3 -1
digestor.py CHANGED
@@ -21,6 +21,7 @@ from scrape_sources import stub as stb
21
  class Digestor:
22
  timer: Timer
23
  cache: bool = True
 
24
  text: str = field(default="no_digest")
25
  stubs: List = field(default_factory=list)
26
  # For clarity.
@@ -45,7 +46,8 @@ class Digestor:
45
  word_limit: int = 400
46
  SUMMARIZATION_PARAMETERS = {
47
  "do_sample": False,
48
- "use_cache": cache
 
49
  }
50
 
51
  # Inference parameters
 
21
  class Digestor:
22
  timer: Timer
23
  cache: bool = True
24
+ temp: float = 1.0
25
  text: str = field(default="no_digest")
26
  stubs: List = field(default_factory=list)
27
  # For clarity.
 
46
  word_limit: int = 400
47
  SUMMARIZATION_PARAMETERS = {
48
  "do_sample": False,
49
+ "use_cache": cache ,
50
+ "temperature":temp,
51
  }
52
 
53
  # Inference parameters