xeal
xeal
type: help. as title.
**Describe** Model I am using (UniLM, MiniLM, LayoutLM ...): UniLM
使用 https://huggingface.co/baichuan-inc/Baichuan2-13B-Chat/tree/v2.0 下载的模型，然后用以下代码load模型 ``` import torch from transformers import AutoModelForCausalLM, AutoTokenizer from transformers.generation.utils import GenerationConfig model_dir = "./Baichuan2-13B-Chat-v2.0" tokenizer = AutoTokenizer.from_pretrained(model_dir, revision="v2.0", use_fast=False, trust_remote_code=True) model = AutoModelForCausalLM.from_pretrained(model_dir, revision="v2.0", device_map="auto", torch_dtype=torch.bfloat16,...