Skip to content

Commit 2e42e7c

Browse files
remove model path from get_cpu_model args
1 parent 32c1e6c commit 2e42e7c

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/models/test_decoders.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -527,7 +527,7 @@ def _get_common_model_kwargs(is_gptq, model_path):
527527

528528
# NOTE micro_model_state_dict should be None if USE_MICRO_MODELS is true
529529
# Otherwise it should be model.state_dict() where model is the AIU model
530-
def _get_cpu_model(model_path, is_gptq, is_fp8, micro_model_state_dict=None, **kwargs):
530+
def _get_cpu_model(is_gptq, is_fp8, micro_model_state_dict=None, **kwargs):
531531
# prepare the cpu model
532532
validation_model = get_model(
533533
device_type="cpu",
@@ -871,7 +871,6 @@ def test_common_shapes(
871871
)
872872

873873
validation_model = _get_cpu_model(
874-
model_path,
875874
is_gptq,
876875
is_fp8,
877876
micro_model_state_dict=model.state_dict() if USE_MICRO_MODELS else None,

0 commit comments

Comments
 (0)