Feature: Adding model names for LLAMA2 and Mistral.

BinSquare · Nov 9, 2023 · 309bab4 · 309bab4
1 parent 268f718
commit 309bab4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/completion.go b/completion.go
@@ -56,6 +56,9 @@ const (
 	GPT3Ada002            = "ada-002"
 	GPT3Babbage           = "babbage"
 	GPT3Babbage002        = "babbage-002"
+
+	LLAMA2DOT70B         = "meta-llama/Llama-2-70b-chat-hf"
+	MISTRALInstructDOT7B = "mistralai/Mistral-7B-Instruct-v0.1"
 )
 
 // Codex Defines the models provided by OpenAI.