我们是谁:Virtruv
该模型尝试专注于韩语中的数学模型进行学习。
基础模型:'beomi/OPEN-SOLAR-KO-10.7B'
数据集:
traintogpb/aihub-koen-translation-integrated-tiny-100k
kyujinpy/KOR-gugugu-platypus-set
GAIR/MathPile:对以下数据集进行采样后直接翻译而得。
您可以这样使用该模型:
import argparse
import torch
from openmind import is_torch_npu_available, pipeline
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument(
"--model_name_or_path",
type=str,
help="Path to model",
default=None,
)
args = parser.parse_args()
return args
def main():
args = parse_args()
model_path = args.model_name_or_path
if is_torch_npu_available():
device = "npu:0"
else:
device = "cpu"
generator = pipeline("text-generation", model=model_path, device_map="auto")
output = generator("자기소개 좀 해주세요.")
print(output)
if __name__ == "__main__":
main()