Update codellama-7b.yaml

Signed-off-by: Ettore Di Giacinto <mudler@users.noreply.github.com>
mudler · Jan 30, 2024 · 555bc02 · 555bc02
1 parent c1bae1e
commit 555bc02
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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