Spaces:
Runtime error
Runtime error
Fixed bug with default prompt style not being valid
Browse files- 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 = "
|
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
|