Skip to content

Commit 1895c11

Browse files
committed
Rename postfix to suffix to match upstream
1 parent 0e9f227 commit 1895c11

File tree

3 files changed

+5
-4
lines changed

3 files changed

+5
-4
lines changed

examples/low_level_api/Chat.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@ def env_or_def(env, default):
6363
interactive=True,
6464
antiprompt=[f"{USER_NAME}:"],
6565
input_prefix=" ",
66+
input_suffix=f"{AI_NAME}:",
6667
prompt=prompt,
6768
)
6869

examples/low_level_api/common.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@ class GptParams:
3737
prompt: str = ""
3838
path_session: str = ""
3939
input_prefix: str = " "
40+
input_suffix: str = ""
4041
antiprompt: List[str] = field(default_factory=list)
4142

4243
lora_adapter: str = ""
@@ -64,7 +65,6 @@ class GptParams:
6465
# Set to "\nUser:" etc.
6566
# This is an alternative to input_prefix which always adds it, so it potentially duplicates "User:""
6667
fix_prefix: str = ""
67-
output_postfix: str = ""
6868
input_echo: bool = True,
6969

7070
# Default instructions for Alpaca
@@ -110,6 +110,7 @@ def gpt_params_parse(argv = None):
110110
parser.add_argument("-f", "--file", type=str, default=None, help="file containing initial prompt to load",dest="file")
111111
parser.add_argument("--session", type=str, default=None, help="file to cache model state in (may be large!)",dest="path_session")
112112
parser.add_argument("--in-prefix", type=str, default="", help="string to prefix user inputs with", dest="input_prefix")
113+
parser.add_argument("--in-suffix", type=str, default="", help="append to input", dest="input_suffix")
113114
parser.add_argument(
114115
"-r",
115116
"--reverse-prompt",
@@ -158,7 +159,6 @@ def gpt_params_parse(argv = None):
158159

159160
#Custom args
160161
parser.add_argument("--fix-prefix", type=str, default="", help="append to input when generated n_predict tokens", dest="fix_prefix")
161-
parser.add_argument("--out-postfix", type=str, default="", help="append to input", dest="output_postfix")
162162
parser.add_argument("--input-noecho", action="store_false", help="dont output the input", dest="input_echo")
163163

164164
parser.add_argument(

examples/low_level_api/low_level_api_chat_cpp.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -527,8 +527,8 @@ def interact(self):
527527
self.input(self.read_input())
528528
else:
529529
print(self.params.input_prefix, end="")
530-
self.input(f"{self.params.input_prefix}{self.read_input()}{self.params.output_postfix}")
531-
print(self.params.output_postfix,end="")
530+
self.input(f"{self.params.input_prefix}{self.read_input()}{self.params.input_suffix}")
531+
print(self.params.input_suffix,end="")
532532
self.set_color(CONSOLE_COLOR_DEFAULT)
533533

534534
try:

0 commit comments

Comments
 (0)