alfraser commited on
Commit
190ec66
·
1 Parent(s): 53169ab

Fixed bug with default prompt style not being valid

Browse files
Files changed (1) hide show
  1. src/architectures.py +1 -1
src/architectures.py CHANGED
@@ -437,7 +437,7 @@ class HFInferenceEndpoint(ArchitectureComponent):
437
  inference endpoint on HuggingFace
438
  """
439
  def __init__(self, endpoint_url: str, model_name: str, system_prompt: str, max_new_tokens: int,
440
- temperature: float = 1.0, prompt_style: str = "full"):
441
  self.endpoint_url: str = endpoint_url
442
  self.prompt_style = prompt_style
443
  self.model_name: str = model_name
 
437
  inference endpoint on HuggingFace
438
  """
439
  def __init__(self, endpoint_url: str, model_name: str, system_prompt: str, max_new_tokens: int,
440
+ temperature: float = 1.0, prompt_style: str = "multi_line"):
441
  self.endpoint_url: str = endpoint_url
442
  self.prompt_style = prompt_style
443
  self.model_name: str = model_name