Skip to content

Commit 732d75e

Browse files
committed
fix: test_huggingface_build_deploy_invoke_cleanup not accepting anymore
single-segment hub id Signed-off-by: DWarez <dario.salvati@huggingface.co>
1 parent baab0a6 commit 732d75e

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

‎sagemaker-serve/tests/integ/test_huggingface_integration.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
logger = logging.getLogger(__name__)
2626

2727
# Configuration - easily customizable
28-
MODEL_ID = "gpt2" # Small causal language model compatible with DJL 0.36.0 vLLM
28+
MODEL_ID = "openai-community/gpt2" # Small causal language model compatible with DJL 0.36.0 vLLM
2929
MODEL_NAME_PREFIX = "hf-test-model"
3030
ENDPOINT_NAME_PREFIX = "hf-test-endpoint"
3131

@@ -130,4 +130,4 @@ def cleanup_resources(core_model, core_endpoint):
130130
core_endpoint.delete()
131131
core_endpoint_config.delete()
132132

133-
logger.info("Model and Endpoint Successfully Deleted!")
133+
logger.info("Model and Endpoint Successfully Deleted!")

0 commit comments

Comments
 (0)