Skip to content

Commit 10f3e10

Browse files
committed
update makefile and requirements
1 parent d0cd413 commit 10f3e10

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

huggingface-sagemaker/Makefile

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,12 @@
1-
stack_name ?= nlp_template_stack
1+
stack_name ?= huggingface_local_stack
22
setup:
33
pip install -r requirements.txt
4-
zenml integration install pytorch mlflow huggingface aws s3 kubeflow slack github -y
4+
zenml integration install pytorch mlflow huggingface aws s3 kubeflow slack github evidently -y
55

66
install-stack:
77
@echo "Specify stack name [$(stack_name)]: " && read input && [ -n "$$input" ] && stack_name="$$input" || stack_name="$(stack_name)" && \
88
zenml experiment-tracker register -f mlflow mlflow_local_$${stack_name} && \
9+
zenml data-validator register -f evidently evidently_local_$${stack_name} && \
910
zenml model-registry register -f mlflow mlflow_local_$${stack_name} && \
1011
zenml model-deployer register -f mlflow mlflow_local_$${stack_name} && \
1112
zenml stack register -a default -o default -r mlflow_local_$${stack_name} \

huggingface-sagemaker/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
torch==2.1.1
22
torchvision==0.16.1
3-
zenml[server]==0.72.0
3+
zenml[server]>=0.72.0
44
sagemaker==2.117.0
55
cuda-python==12.3.0
66
nvidia-cuda-cupti-cu12==12.1.105

0 commit comments

Comments
 (0)