Skip to content

make test_networks.py not rely on HF downloads #2202

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
t-vi merged 6 commits into from
Jun 6, 2025
Merged

make test_networks.py not rely on HF downloads #2202

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
e85b2a3
modif tests to not rely on hf cache
KaelanDt Jun 6, 2025
3253239
rest
KaelanDt Jun 6, 2025
5a93f20
rename function
KaelanDt Jun 6, 2025
b531535
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jun 6, 2025
36e5eea
qwen2
KaelanDt Jun 6, 2025
91e8887
revert to minimal config for tests
KaelanDt Jun 6, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
[pre-commit.ci] auto fixes from pre-commit.com hooks 
for more information, see https://pre-commit.ci
  • Loading branch information
@pre-commit-ci
pre-commit-ci[bot] committed Jun 6, 2025
commit b531535b823a26a01aa78c6c696e35cabae70a1b
5 changes: 4 additions & 1 deletion thunder/tests/test_networks.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@

from transformers.models.qwen2 import Qwen2Config, Qwen2ForCausalLM
from transformers.models.phi3 import Phi3Config, Phi3ForCausalLM

#
# nanoGPT tests
#
Expand Down Expand Up @@ -410,10 +411,12 @@ def _get_model_config_pairs():

def phi3():
from transformers.models.phi3 import Phi3ForCausalLM, Phi3Config

return Phi3ForCausalLM, Phi3Config

def qwen2():
from transformers.models.qwen2 import Qwen2ForCausalLM, Qwen2Config

return Qwen2ForCausalLM, Qwen2Config

return [phi3(), qwen2()]
Expand All @@ -434,7 +437,7 @@ def test_hf_for_nemo(model_cls, config_cls):
max_position_embeddings=128,
use_cache=True,
tie_word_embeddings=False,
pad_token_id=15
pad_token_id=15,
)

with torch.device("cuda"):
Expand Down
Loading