Use Qwen 2.5 for decoder-only test

jncraton · Jan 24, 2025 · 1bd2833 · 1bd2833
1 parent ddce63d
commit 1bd2833
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/makefile b/makefile
@@ -5,7 +5,7 @@ all: lint test
 test-base:
 	python3 -m doctest -o ELLIPSIS -o NORMALIZE_WHITESPACE languagemodels/*.py
 	env LANGUAGEMODELS_MODEL_LICENSE="apache|mit|bsd" python3 -m doctest -o ELLIPSIS -o NORMALIZE_WHITESPACE languagemodels/__init__.py
-	LANGUAGEMODELS_INSTRUCT_MODEL="LaMini-GPT-124M" python3 -m doctest -o ELLIPSIS -o NORMALIZE_WHITESPACE languagemodels/inference.py
+	LANGUAGEMODELS_INSTRUCT_MODEL="Qwen2.5-0.5B-Instruct" python3 -m doctest -o ELLIPSIS -o NORMALIZE_WHITESPACE languagemodels/inference.py
 
 test: test-base
 	env LANGUAGEMODELS_MAX_RAM=large python3 -m doctest -o ELLIPSIS -o NORMALIZE_WHITESPACE languagemodels/*.py