init project
This commit is contained in:
6
docker.sh
Executable file
6
docker.sh
Executable file
@@ -0,0 +1,6 @@
|
||||
docker stop bi150_tc
|
||||
docker rm bi150_tc
|
||||
|
||||
docker build . -t bi150_text_classification
|
||||
|
||||
docker run --rm -p 17777:8000 -v $mountpath:/model:ro -it --device=/dev/iluvatar0:/dev/iluvatar0 --name bi150_tc -e CONFIG_JSON='{"model_class": "AutoModelForSequenceClassification", "tokenizer_class": "AutoTokenizer", "torch_dtype": "auto"}' bi150_text_classification
|
||||
Reference in New Issue
Block a user