File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
sagemaker-serve/src/sagemaker/serve Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -687,7 +687,7 @@ def _build_for_transformers(self) -> Model:
687687 hf_model_id , self .env_vars .get ("HUGGING_FACE_HUB_TOKEN" )
688688 )
689689 elif isinstance (self .model , str ): # Only set HF_MODEL_ID if model is a string
690- # Get model metadata for task detection (same pattern as _build_for_triton)
690+ # Get model metadata for task detection
691691 hf_model_md = self .get_huggingface_model_metadata (
692692 self .model , self .env_vars .get ("HUGGING_FACE_HUB_TOKEN" )
693693 )
@@ -697,7 +697,7 @@ def _build_for_transformers(self) -> Model:
697697
698698 self .env_vars .update ({"HF_MODEL_ID" : self .model })
699699
700- # Add HuggingFace token if available (same as other methods)
700+ # Add HuggingFace token if available
701701 if self .env_vars .get ("HUGGING_FACE_HUB_TOKEN" ):
702702 self .env_vars ["HF_TOKEN" ] = self .env_vars .get ("HUGGING_FACE_HUB_TOKEN" )
703703
You can’t perform that action at this time.
0 commit comments