7 lines
356 B
Bash
7 lines
356 B
Bash
|
|
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
|