You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
18 lines
2.5 KiB
18 lines
2.5 KiB
import os
|
|
os.environ["CUDA_VISIBLE_DEVICES"] = "2,3"
|
|
import torch
|
|
from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
from transformers.generation.utils import GenerationConfig
|
|
|
|
model_path = "/home/majiahui/project/models-llm/Baichuan-13B-Chat"
|
|
|
|
tokenizer = AutoTokenizer.from_pretrained(model_path, use_fast=False, trust_remote_code=True)
|
|
model = AutoModelForCausalLM.from_pretrained(model_path, device_map="auto", torch_dtype=torch.float16, trust_remote_code=True)
|
|
model.generation_config = GenerationConfig.from_pretrained(model_path)
|
|
messages = []
|
|
# messages.append({"role": "user", "content": "把下面这段翻译成英文\n研究背景：\n\n随着全球能源需求的不断增加和化石燃料的日益枯竭，新能源技术的研究和应用已经成为全球性的热点问题。其中，燃料电池作为一种高效、清洁、可再生的新能源技术，已经成为国际上研究的热点之一。而其中，高温质子交换膜燃料电池（HT-PEMFC）由于具有高能量密度、高效率、低污染等优点，被广泛应用于航空航天、汽车、家庭等领域。\n\n然而，HT-PEMFC在启动过程中存在着一定的困难。由于燃料电池的运行需要一定的温度和湿度条件，因此在启动时需要先加热和湿化电池，以达到运行所需的条件。而传统的加热和湿化方法需要较长的时间，且能量消耗较大，因此需要寻找一种快速启动的方法，以提高燃料电池的运行效率和可靠性。\n\n研究意义：\n\n本研究旨在探究一种快速启动HT-PEMFC的方法，以提高其运行效率和可靠性。具体而言，本研究将采用仿真方法，研究在不同的温度和湿度条件下，利用燃料电池的内部能量来快速启动电池的可行性和效果，并分析其影响因素和优化方案。\n\n本研究的意义在于：\n\n1. 提高HT-PEMFC的运行效率和可靠性。采用快速启动方法可以减少启动时间和能量消耗，提高燃料电池的运行效率和可靠性。\n\n2. 探索新的启动方法。传统的加热和湿化方法存在能量消耗大、启动时间长等缺点，因此本研究探索一种新的快速启动方法，为燃料电池的应用提供新的思路和方法。\n\n3. 为燃料电池的应用提供技术支持。燃料电池作为一种新能源技术，其应用前景广阔。本研究为燃料电池的应用提供技术支持，有利于推动其产业化和普及。\n"})
|
|
|
|
messages.append({"role": "user", "content": ""})
|
|
|
|
response = model.chat(tokenizer, messages)
|
|
print(response)
|