Archives/reflexion-human-eval
2
0
Fork 0
mirror of https://github.com/GammaTauAI/reflexion-human-eval synced 2024-11-16 00:12:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

temp fix

Browse Source
This commit is contained in:
cassanof 2023-06-18 04:08:33 -07:00
parent a9d34708ad
commit b9d2c54114
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
programming_runs/generators/model.py
View File

@ -116,6 +116,10 @@ class StarChat(ModelBase):
self.is_chat = True
def generate_chat(self, system_message: str, user_message: str, max_tokens=1024, temperature=0.2, num_comps=1) -> Union[List[str], str]:
# NOTE: HF does not like temp of 0.0.
if temperature < 0.0001:
temperature = 0.0001
prompt = self.template.format(
system=system_message, query=user_message)
outputs = self.pipe(