green commited on
Commit
8a15221
1 Parent(s): 33be678

Update digestor.py

Browse files
Files changed (1) hide show
  1. digestor.py +1 -3
digestor.py CHANGED
@@ -21,7 +21,6 @@ from scrape_sources import stub as stb
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,8 +45,7 @@ class Digestor:
46
  word_limit: int = 400
47
  SUMMARIZATION_PARAMETERS = {
48
  "do_sample": False,
49
- "use_cache": cache ,
50
- "temperature":temp,
51
  }
52
 
53
  # Inference parameters
 
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
  word_limit: int = 400
46
  SUMMARIZATION_PARAMETERS = {
47
  "do_sample": False,
48
+ "use_cache": cache,
 
49
  }
50
 
51
  # Inference parameters