@@ -37,6 +37,7 @@ class GptParams:
37
37
prompt : str = ""
38
38
path_session : str = ""
39
39
input_prefix : str = " "
40
+ input_suffix : str = ""
40
41
antiprompt : List [str ] = field (default_factory = list )
41
42
42
43
lora_adapter : str = ""
@@ -64,7 +65,6 @@ class GptParams:
64
65
# Set to "\nUser:" etc.
65
66
# This is an alternative to input_prefix which always adds it, so it potentially duplicates "User:""
66
67
fix_prefix : str = ""
67
- output_postfix : str = ""
68
68
input_echo : bool = True ,
69
69
70
70
# Default instructions for Alpaca
@@ -110,6 +110,7 @@ def gpt_params_parse(argv = None):
110
110
parser .add_argument ("-f" , "--file" , type = str , default = None , help = "file containing initial prompt to load" ,dest = "file" )
111
111
parser .add_argument ("--session" , type = str , default = None , help = "file to cache model state in (may be large!)" ,dest = "path_session" )
112
112
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" )
113
114
parser .add_argument (
114
115
"-r" ,
115
116
"--reverse-prompt" ,
@@ -158,7 +159,6 @@ def gpt_params_parse(argv = None):
158
159
159
160
#Custom args
160
161
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" )
162
162
parser .add_argument ("--input-noecho" , action = "store_false" , help = "dont output the input" , dest = "input_echo" )
163
163
164
164
parser .add_argument (
0 commit comments