From 8be974d1a45924412ab86f0169eec7a150efd9e5 Mon Sep 17 00:00:00 2001 From: Arpit Jasapara Date: Fri, 28 Jun 2024 00:40:57 -0700 Subject: [PATCH] modify test model --- internal/acceptance/model_serving_test.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/internal/acceptance/model_serving_test.go b/internal/acceptance/model_serving_test.go index d791067e19..2204d57fdb 100644 --- a/internal/acceptance/model_serving_test.go +++ b/internal/acceptance/model_serving_test.go @@ -97,7 +97,7 @@ func TestUcAccModelServingProvisionedThroughput(t *testing.T) { config { served_entities{ name = "pt_model" - entity_name = "system.ai.mistral_7b_instruct_v0_1" + entity_name = "system.ai.mistral_7b_instruct_v0_2" entity_version = "1" min_provisioned_throughput = 0 max_provisioned_throughput = 970 @@ -118,7 +118,7 @@ func TestUcAccModelServingProvisionedThroughput(t *testing.T) { config { served_entities{ name = "pt_model" - entity_name = "system.ai.mistral_7b_instruct_v0_1" + entity_name = "system.ai.mistral_7b_instruct_v0_2" entity_version = "1" min_provisioned_throughput = 970 max_provisioned_throughput = 1940 @@ -139,7 +139,7 @@ func TestUcAccModelServingProvisionedThroughput(t *testing.T) { config { served_entities{ name = "pt_model" - entity_name = "system.ai.mistral_7b_instruct_v0_1" + entity_name = "system.ai.mistral_7b_instruct_v0_2" entity_version = "1" min_provisioned_throughput = 0 max_provisioned_throughput = 1940