lingyuzeng/cdc_dockerfile
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
lingyuzeng
2024-07-18 15:22:19 +08:00
parent b7584b2a04
commit 5f18f0d3c1
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
finetune/Dockerfile.ngc
View File

@@ -79,7 +79,12 @@ EOT
RUN <<EOT RUN <<EOT
#!/bin/bash #!/bin/bash
pip install -v -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers 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 pip install git+https://github.com/huggingface/transformers
python -c "from transformers import pipeline; print(pipeline('sentiment-analysis')('I love you'))"
EOT EOT
RUN <<EOT RUN <<EOT