riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 03:01:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Phi3 tests: fix typing for Python 3.8 (#32388)

Browse Source 
fix phi
This commit is contained in:
Raushan Turganbay 2024-08-05 11:58:42 +05:00 committed by GitHub
parent 05ae3a300d
commit 3bb646a54f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/phi3/test_modeling_phi3.py
View File

@ -16,6 +16,7 @@
"""Testing suite for the PyTorch Phi-3 model.""" """Testing suite for the PyTorch Phi-3 model."""
import unittest import unittest
from typing import List
from parameterized import parameterized from parameterized import parameterized
@ -69,7 +70,7 @@ if is_torch_available():
).logits ).logits
@staticmethod @staticmethod
def generate(model: Phi3ForCausalLM, prompt_tokens: torch.LongTensor, max_seq_len: int) -> list[int]: def generate(model: Phi3ForCausalLM, prompt_tokens: torch.LongTensor, max_seq_len: int) -> List[int]:
model = Phi3MiniWithStaticCache(model, 1, max_seq_len + prompt_tokens.shape[-1]) model = Phi3MiniWithStaticCache(model, 1, max_seq_len + prompt_tokens.shape[-1])
response_tokens = [] response_tokens = []