bot: load the pythia model by default

csris · csris · commit 1ed1d6e45c35 · 2023-03-30T20:46:35.000Z
diff --git a/inference/bot.py b/inference/bot.py
@@ -202,7 +202,7 @@ def main():
     )
     parser.add_argument(
         '--model',
-        default=f"{INFERENCE_DIR}/../huggingface_models/GPT-NeoXT-Chat-Base-20B",
+        default=f"{INFERENCE_DIR}/../huggingface_models/Pythia-Chat-Base-7B",
         help='name/path of the model'
     )
     parser.add_argument(

Original file line number	Diff line number	Diff line change
`@@ -202,7 +202,7 @@ def main():`
`202`	`202`	`)`
`203`	`203`	`parser.add_argument(`
`204`	`204`	`'--model',`
`205`		`- default=f"{INFERENCE_DIR}/../huggingface_models/GPT-NeoXT-Chat-Base-20B",`
	`205`	`+ default=f"{INFERENCE_DIR}/../huggingface_models/Pythia-Chat-Base-7B",`
`206`	`206`	`help='name/path of the model'`
`207`	`207`	`)`
`208`	`208`	`parser.add_argument(`