Using llama3.2:1b to reduce the size of the docker image to download (#5

)
AtomicJar · Oct 28, 2024 · 6fd342c · 6fd342c
1 parent e37aa7e
commit 6fd342c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/test/java/com/example/demogenai/ContainersConfiguration.java b/src/test/java/com/example/demogenai/ContainersConfiguration.java
@@ -15,7 +15,7 @@ public class ContainersConfiguration {
 	@ServiceConnection
 	@RestartScope
 	OllamaContainer ollama() {
-		return new OllamaContainer(DockerImageName.parse("ilopezluna/evaluator-llama3.1:0.3.11-8b")
+		return new OllamaContainer(DockerImageName.parse("ilopezluna/spring-ai-demo-models:llama3.2-1b")
 			.asCompatibleSubstituteFor("ollama/ollama")) {
 		};
 	}

diff --git a/src/test/resources/application.properties b/src/test/resources/application.properties
@@ -4,7 +4,7 @@ spring.ai.ollama.embedding.options.top-k=2
 spring.ai.ollama.embedding.options.seed=42
 spring.ai.ollama.embedding.options.temperature=0.0
 
-spring.ai.ollama.chat.options.model=llama3.1:8b
+spring.ai.ollama.chat.options.model=llama3.2:1b
 spring.ai.ollama.chat.options.top-k=1
 spring.ai.ollama.chat.options.seed=42
 spring.ai.ollama.chat.options.temperature=0.0