Merge pull request #52 from thilohuellmann/patch-1

philschmid · web-flow · commit 5b25d123cd4a · 2022-01-18T11:53:41.000+01:00
Hint at correct env var name in error message
diff --git a/src/sagemaker_huggingface_inference_toolkit/handler_service.py b/src/sagemaker_huggingface_inference_toolkit/handler_service.py
@@ -107,7 +107,7 @@ def load(self, model_dir):
             hf_pipeline = get_pipeline(task=task, model_dir=model_dir, device=self.device)
         else:
             raise ValueError(
-                f"You need to define one of the following {list(SUPPORTED_TASKS.keys())} as env 'TASK'.", 403
+                f"You need to define one of the following {list(SUPPORTED_TASKS.keys())} as env 'HF_TASK'.", 403
             )
         return hf_pipeline
 

Original file line number	Diff line number	Diff line change
`@@ -107,7 +107,7 @@ def load(self, model_dir):`
`107`	`107`	`hf_pipeline = get_pipeline(task=task, model_dir=model_dir, device=self.device)`
`108`	`108`	`else:`
`109`	`109`	`raise ValueError(`
`110`		`- f"You need to define one of the following {list(SUPPORTED_TASKS.keys())} as env 'TASK'.", 403`
	`110`	`+ f"You need to define one of the following {list(SUPPORTED_TASKS.keys())} as env 'HF_TASK'.", 403`
`111`	`111`	`)`
`112`	`112`	`return hf_pipeline`
`113`	`113`