Hansimov commited on
Commit
eb00725
1 Parent(s): 8ba223c

:boom: [Fix] temperature must be strictly positive

Browse files
Files changed (1) hide show
  1. networks/message_streamer.py +7 -3
networks/message_streamer.py CHANGED
@@ -43,7 +43,10 @@ class MessageStreamer:
43
  line = line.decode("utf-8")
44
  line = re.sub(r"data:\s*", "", line)
45
  data = json.loads(line)
46
- content = data["token"]["text"]
 
 
 
47
  return content
48
 
49
  def count_tokens(self, text):
@@ -76,8 +79,9 @@ class MessageStreamer:
76
 
77
  if temperature is None or temperature < 0:
78
  temperature = 0.0
79
- # temperature must be positive for HF LLM models
80
- temperature = min(temperature, 0.01)
 
81
 
82
  token_limit = (
83
  self.TOKEN_LIMIT_MAP[self.model]
 
43
  line = line.decode("utf-8")
44
  line = re.sub(r"data:\s*", "", line)
45
  data = json.loads(line)
46
+ try:
47
+ content = data["token"]["text"]
48
+ except:
49
+ logger.err(data)
50
  return content
51
 
52
  def count_tokens(self, text):
 
79
 
80
  if temperature is None or temperature < 0:
81
  temperature = 0.0
82
+ # temperature must be positive and <= 1 for HF LLM models
83
+ temperature = max(temperature, 0.01)
84
+ temperature = min(temperature, 1)
85
 
86
  token_limit = (
87
  self.TOKEN_LIMIT_MAP[self.model]