From 555bc0266530ceaa4edb3624fe970c88c497ffab Mon Sep 17 00:00:00 2001
From: Ettore Di Giacinto <mudler@users.noreply.github.com>
Date: Tue, 30 Jan 2024 11:36:20 +0100
Subject: [PATCH] Update codellama-7b.yaml

Signed-off-by: Ettore Di Giacinto <mudler@users.noreply.github.com>
---
 embedded/models/codellama-7b.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/embedded/models/codellama-7b.yaml b/embedded/models/codellama-7b.yaml
index 151f7515..d9b5c62c 100644
--- a/embedded/models/codellama-7b.yaml
+++ b/embedded/models/codellama-7b.yaml
@@ -1,5 +1,6 @@
 name: codellama-7b
 backend: transformers
+type: AutoModelForCausalLM
 parameters:
   model: codellama/CodeLlama-7b-hf
   temperature: 0.2