Views
No views yet
Qwen/Qwen-7B-Chattrain_qwen7b_lora.pytest_compare.py1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4model_name = "Josh1207/qwen7b-alpaca-lora"
5
6tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
7base_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen-7B-Chat", trust_remote_code=True)
8model = PeftModel.from_pretrained(base_model, model_name)
9
10prompt = "指令: 请介绍一下你自己。"
11inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
12outputs = model.generate(**inputs, max_new_tokens=512)
13print(tokenizer.decode(outputs[0], skip_special_tokens=True))Qwen/Qwen-7B-Chatpeft)train_qwen7b_lora.pytest_compare.py1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4model_name = "Josh1207/qwen7b-alpaca-lora"
5
6tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
7base_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen-7B-Chat", trust_remote_code=True)
8model = PeftModel.from_pretrained(base_model, model_name)
9
10prompt = "指令: 请介绍一下你自己。"
11inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
12outputs = model.generate(**inputs, max_new_tokens=512)
13print(tokenizer.decode(outputs[0], skip_special_tokens=True))