From bfae79a63466e16f64c520a3f99b40dbc29d66b0 Mon Sep 17 00:00:00 2001 From: Mads Henrichsen Date: Sat, 30 Dec 2023 12:47:50 +0100 Subject: [PATCH] trust --- examples/yayi2-30b/qlora.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/yayi2-30b/qlora.yml b/examples/yayi2-30b/qlora.yml index 55b213d2e..90172cdad 100644 --- a/examples/yayi2-30b/qlora.yml +++ b/examples/yayi2-30b/qlora.yml @@ -2,6 +2,7 @@ base_model: wenge-research/yayi2-30b model_type: AutoModelForCausalLM tokenizer_type: AutoTokenizer is_mistral_derived_model: false +trust_remote_code: true load_in_8bit: false load_in_4bit: true