diff --git a/examples/yayi2-30b/qlora.yml b/examples/yayi2-30b/qlora.yml index d1ab81bdc..a6500104e 100644 --- a/examples/yayi2-30b/qlora.yml +++ b/examples/yayi2-30b/qlora.yml @@ -3,6 +3,7 @@ model_type: AutoModelForCausalLM tokenizer_type: AutoTokenizer is_mistral_derived_model: false trust_remote_code: true +model_revision: refs/pr/5 load_in_8bit: false load_in_4bit: true