Update model_config.py
This commit is contained in:
parent
b9d8824fa5
commit
b48e399e9e
|
|
@ -23,7 +23,7 @@ llm_model_dict = {
|
||||||
}
|
}
|
||||||
|
|
||||||
# LLM model name
|
# LLM model name
|
||||||
LLM_MODEL = "chatyuan" #"chatglm-6b"
|
LLM_MODEL = "chatglm-6b"
|
||||||
|
|
||||||
# Use p-tuning-v2 PrefixEncoder
|
# Use p-tuning-v2 PrefixEncoder
|
||||||
USE_PTUNING_V2 = False
|
USE_PTUNING_V2 = False
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue