Hi, I have the source code for a custom data cleaning library; the code is private to my company. I usually install it from local source using something ‘pip install ./cleanse’. I know the requirements.txt file can be used to specify package versions for public libraries like pandas and NumPy but h…

does the model_uri contain a code/inference.py ? If so you need to put the requirements.txt / ./cleanse into the code/ directory.

Install custom python libraries in HuggingFace DLC

Amazon SageMaker

philschmid August 8, 2022, 2:03pm 4

Sorry, i am not sure what you mean with Is there a place I could leverage the inference.py file inside the code/ directory in the model.tar.gz file?

Topic		Replies	Views
Inference Toolkit - Init and default template for custom inference Amazon SageMaker	12	2139	October 4, 2021
HuggingFaceModel ignores code directory Amazon SageMaker	2	16	June 17, 2025
Adding missing packages in HF DLC Amazon SageMaker	1	353	September 14, 2023
Loading inference.py separately from model.tar.gz Amazon SageMaker	4	1862	June 5, 2023
Infer on sagemaker with custom pipeline Amazon SageMaker	2	498	September 14, 2023

Install custom python libraries in HuggingFace DLC

Related topics