chore: 移除冗余代码行以提高可读性

This commit is contained in:
songsenand 2026-04-09 22:19:56 +08:00
commit 8b0beeb56c
1 changed files with 1 additions and 1 deletions

View File

@ -80,11 +80,11 @@ sample = {
"pinyin_ids": pinyin_ids,
}
model = InputMethodEngine(pinyin_vocab_size=30, compile=False)
checkpoint = torch.load("/home/songsenand/下载/best_model.pt", map_location="cpu")
model.load_state_dict(checkpoint["model_state_dict"])
input_ids = sample["input_ids"]
token_type_ids = sample["token_type_ids"]
attention_mask = sample["attention_mask"]