From 13c59cc97bdc4800ddee6c06a1c16e1ca2024208 Mon Sep 17 00:00:00 2001
From: Carl-Robert Linnupuu <carlrobertoh@gmail.com>
Date: Tue, 7 May 2024 18:20:06 +0300
Subject: [PATCH] fix: build

---
 .../codegpt/settings/service/llama/LlamaSettings.java         | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/main/java/ee/carlrobert/codegpt/settings/service/llama/LlamaSettings.java b/src/main/java/ee/carlrobert/codegpt/settings/service/llama/LlamaSettings.java
index fa3fef84..926fd021 100644
--- a/src/main/java/ee/carlrobert/codegpt/settings/service/llama/LlamaSettings.java
+++ b/src/main/java/ee/carlrobert/codegpt/settings/service/llama/LlamaSettings.java
@@ -34,10 +34,10 @@ public class LlamaSettings implements PersistentStateComponent<LlamaSettingsStat
       this.state.setHuggingFaceModel(HuggingFaceModel.CODE_LLAMA_7B_Q4);
     }
     if (this.state.getRemoteModelInfillPromptTemplate() == null) {
-      this.state.setRemoteModelInfillPromptTemplate(InfillPromptTemplate.LLAMA);
+      this.state.setRemoteModelInfillPromptTemplate(InfillPromptTemplate.CODE_LLAMA);
     }
     if (this.state.getLocalModelPromptTemplate() == null) {
-      this.state.setLocalModelInfillPromptTemplate(InfillPromptTemplate.LLAMA);
+      this.state.setLocalModelInfillPromptTemplate(InfillPromptTemplate.CODE_LLAMA);
     }
   }