@@ -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 (
0 commit comments