Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3torch.manual_seed(0)
4
5path = 'openbmb/MiniCPM-MoE-8x2B'
6tokenizer = AutoTokenizer.from_pretrained(path)
7model = AutoModelForCausalLM.from_pretrained(path, torch_dtype=torch.bfloat16, device_map='cuda', trust_remote_code=True)
8
9responds, history = model.chat(tokenizer, "山东省最高的山是哪座山, 它比黄山高还是矮?差距多少?", temperature=0.8, top_p=0.8)
10print(responds)