From 46894f9ec4b48d4e14cd2ca6c7e78446c2d7164d Mon Sep 17 00:00:00 2001
From: Amith Koujalgi <amith.koujalgi@razorthink.com>
Date: Fri, 27 Oct 2023 17:31:45 +0530
Subject: [PATCH] init

---
 src/main/java/io/github/amithkoujalgi/ollama4j/OllamaModel.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/java/io/github/amithkoujalgi/ollama4j/OllamaModel.java b/src/main/java/io/github/amithkoujalgi/ollama4j/OllamaModel.java
index fe4c541..bbf6954 100644
--- a/src/main/java/io/github/amithkoujalgi/ollama4j/OllamaModel.java
+++ b/src/main/java/io/github/amithkoujalgi/ollama4j/OllamaModel.java
@@ -1,7 +1,7 @@
 package io.github.amithkoujalgi.ollama4j;
 
 public enum OllamaModel {
-    LLAMA2("llama2"), MISTRAL("mistral"), MEDLLAMA2("medllama2");
+    LLAMA2("llama2"), MISTRAL("mistral"), MEDLLAMA2("medllama2"), CODELLAMA("codellama"), VICUNA("vicuna"), ORCAMINI("orca-mini"), sqlcoder("sqlcoder"), WIZARDMATH("wizard-math");
 
     private final String model;