Skip to content

Commit dd355c1

Browse files
remove model path from get_cpu_model args
Signed-off-by: Alex-Brooks <Alex.Brooks@ibm.com>
1 parent dcfa628 commit dd355c1

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
@@ -542,7 +542,7 @@ def _get_common_model_kwargs(is_gptq, model_path):
542542

543543
# NOTE micro_model_state_dict should be None if USE_MICRO_MODELS is true
544544
# Otherwise it should be model.state_dict() where model is the AIU model
545-
def _get_cpu_model(model_path, is_gptq, is_fp8, micro_model_state_dict=None, **kwargs):
545+
def _get_cpu_model(is_gptq, is_fp8, micro_model_state_dict=None, **kwargs):
546546
# prepare the cpu model
547547
validation_model = get_model(
548548
device_type="cpu",
@@ -905,7 +905,6 @@ def test_common_shapes(
905905
)
906906

907907
validation_model = _get_cpu_model(
908-
model_path,
909908
is_gptq,
910909
is_fp8,
911910
micro_model_state_dict=model.state_dict() if USE_MICRO_MODELS else None,

0 commit comments

Comments
 (0)