From 5f18f0d3c162c9233f805e1f1121697d9bbc5ab3 Mon Sep 17 00:00:00 2001
From: lingyuzeng <pylyzeng@gmail.com>
Date: Thu, 18 Jul 2024 15:22:19 +0800
Subject: [PATCH] update

---
 finetune/Dockerfile.ngc | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/finetune/Dockerfile.ngc b/finetune/Dockerfile.ngc
index 3eb4e6c..4fb183b 100644
--- a/finetune/Dockerfile.ngc
+++ b/finetune/Dockerfile.ngc
@@ -79,7 +79,12 @@ EOT
 RUN <<EOT
 #!/bin/bash
 pip install -v -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers
+EOT
+
+RUN <<EOT
+#!/bin/bash
 pip install git+https://github.com/huggingface/transformers
+python -c "from transformers import pipeline; print(pipeline('sentiment-analysis')('I love you'))"
 EOT
 
 RUN <<EOT