Skip to content

Commit 86b59d6

Browse files
authored
Update ov_phi3_vision.py
Related to openvinotoolkit/openvino_notebooks#2374
1 parent ee5d5e9 commit 86b59d6

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/embeddedllm/backend/ov_phi3_vision.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -322,7 +322,7 @@ def __init__(self, model_dir, device):
322322
self.model = core.read_model(model_dir / "language_model.xml")
323323
self.image_embed = core.compile_model(model_dir / "image_embed.xml", device)
324324
self.img_projection = core.compile_model(model_dir / "img_projection.xml", device)
325-
self.embed_tokem = core.compile_model(model_dir / "embed_token.xml", device)
325+
self.embed_token = core.compile_model(model_dir / "embed_token.xml", device)
326326
self.input_names = {key.get_any_name(): idx for idx, key in enumerate(self.model.inputs)}
327327
self.output_names = {key.get_any_name(): idx for idx, key in enumerate(self.model.outputs)}
328328
compiled_model = core.compile_model(self.model, device)
@@ -573,7 +573,7 @@ def vision_embed_tokens(
573573
select = True
574574
input_ids.clamp_min_(0).clamp_max_(self.config.vocab_size)
575575

576-
hidden_states = torch.from_numpy(self.embed_tokem(input_ids)[0])
576+
hidden_states = torch.from_numpy(self.embed_token(input_ids)[0])
577577
if select:
578578
if hd_transform:
579579
idx = 0
@@ -587,4 +587,4 @@ def vision_embed_tokens(
587587
hidden_states[positions[idx, 0], positions[idx, 1] : positions[idx, 1] + cnt] = img_set_tensor[i * cnt : (i + 1) * cnt]
588588
idx += cnt
589589

590-
return hidden_states
590+
return hidden_states

0 commit comments

Comments
 (0)