Views
No views yet
1import torch
2import torch.nn.functional as F
3from transformers import AutoModelForCausalLM, AutoTokenizer
4from transformers.generation import GenerationConfig
5
6model_path = 'CjangCjengh/WN-zh-vi-sim-v0.2'
7tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
8model = AutoModelForCausalLM.from_pretrained(model_path, device_map='auto', trust_remote_code=True).eval()
9model.generation_config = GenerationConfig.from_pretrained(model_path, trust_remote_code=True)
10
11yes_id = tokenizer.encode('Yes')[0]
12no_id = tokenizer.encode('No')[0]
13
14zh_text = '''听完了倒计时，封不觉很快便进入了游戏的世界中。他置身在一个和电梯十分相像的环境里，但门的旁边没有楼层按钮。右手边的墙壁上是个三十二寸的触摸屏。正对门的那一整面墙是一块完整的镜面，可以将玩家从头照到脚。
15封不觉看了看镜子中自己的形象，此刻他成了个超高像素的三维CG人物。身上的服装变成了黑色的长袖T恤和长裤，没什么特别之处。游戏人物的脸和现实中自己的长相几乎一样，一米八不到，乱糟糟的头发，面部线条略显阴柔。
16在第四代光脑技术的支持下，神经连接游戏中的玩家可以获得足以乱真的五感体验，但游戏画面依然局限于CG影像，而无法让玩家载入到类似真人实景拍摄效果下的环境。'''
17
18vi_text = '''Nghe hết thời gian đếm ngược, Phong Bất Giác nhanh chóng đăng nhập vào giao diện của game. Anh ở trong một không gian rất giống với thang máy, nhưng bên cạnh cửa lại không có nút số tầng. Trên bức tường bên tay phải có màn hình cảm ứng ba mươi hai inch. Bức tường đối diện cửa là một tấm kính nguyên vẹn, có thể soi từ đầu đến chân của người chơi.
19Phong Bất Giác nhìn hình tượng của mình trong gương. Lúc này anh đã biến thành một nhân vật CG 3D độ phân giải cực cao. Quần áo trên người biến thành quần dài và áo T-shirt dài tay màu đen, không có điểm gì đặc biệt. Gương mặt của nhân vật trong game và gương mặt ngoài thật khá giống nhau, chưa đến một mét tám, đầu tóc rối bù, đường nét gương mặt lộ rõ vẻ âm nhu.
20Dưới sự hỗ trợ của máy tính tiên tiến thế hệ bốn, người chơi liên kết thần kinh trong game có thể trải nghiệm năm giác quan ảo, nhưng đồ họa của game hạn chế ở mức hình ảnh CG, không thể cho người chơi môi trường với cảnh thật người thật như khi dùng máy ảnh chụp xuống.'''
21
22zh_text = zh_text.replace('\n', '')
23vi_text = vi_text.replace('\n', ' ')
24
25assert len(zh_text) < 1024
26assert len(vi_text.split()) < 768
27
28messages = [
29 {'role': 'user', 'content': f'下面的中文段落和越南语段落的内容是否完全对应，不存在缺漏？（回答Yes或No）\n\n中文：\n{zh_text}\n\n越南语：{vi_text}'}
30]
31
32text = tokenizer.apply_chat_template(
33 messages,
34 tokenize=False,
35 add_generation_prompt=True
36)
37
38model_inputs = tokenizer([text], return_tensors='pt').to('cuda')
39
40with torch.inference_mode():
41 output = model(**model_inputs, return_dict=True)
42 logits = output.logits[:, -1, [yes_id, no_id]]
43 scores = F.softmax(logits, dim=-1)
44 sim_score = scores[0][0].item()
45
46if sim_score > 0.5:
47 print('Match!')