Add trust_remote_code config
This commit is contained in:
@@ -160,6 +160,8 @@ base_model_config: ./llama-7b-hf
|
|||||||
model_type: AutoModelForCausalLM
|
model_type: AutoModelForCausalLM
|
||||||
# Corresponding tokenizer for the model AutoTokenizer is a good choice
|
# Corresponding tokenizer for the model AutoTokenizer is a good choice
|
||||||
tokenizer_type: AutoTokenizer
|
tokenizer_type: AutoTokenizer
|
||||||
|
# Trust remote code for untrusted source
|
||||||
|
trust_remote_code:
|
||||||
|
|
||||||
# whether you are training a 4-bit quantized model
|
# whether you are training a 4-bit quantized model
|
||||||
load_4bit: true
|
load_4bit: true
|
||||||
|
|||||||
Reference in New Issue
Block a user