Skip to content

Commit d435f64

Browse files
committed
add --force
Signed-off-by: Srikanth Ramakrishna <srikanth.ramakrishna@intel.com>
1 parent ac753b8 commit d435f64

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pytorch/serving/model-archive/mar-test.sh

+2-2
Original file line numberDiff line numberDiff line change
@@ -28,11 +28,11 @@
2828

2929
if [[ "$1" == "cpu" ]]; then
3030
wget https://download.pytorch.org/models/squeezenet1_1-b8a52dc0.pth
31-
torch-model-archiver --model-name squeezenet1_1 --version 1.1 --model-file /home/model-server/model-archive/model.py --serialized-file squeezenet1_1-b8a52dc0.pth --handler image_classifier --export-path /home/model-server/model-store
31+
torch-model-archiver --model-name squeezenet1_1 --version 1.1 --model-file /home/model-server/model-archive/model.py --serialized-file squeezenet1_1-b8a52dc0.pth --handler image_classifier --export-path /home/model-server/model-store --force
3232
rm -rf squeezenet1_1-b8a52dc0.pth
3333
elif [[ "$1" == "xpu" ]]; then
3434
python /home/model-server/model-archive/ipex_squeezenet.py
35-
torch-model-archiver --model-name squeezenet1_1 --version 1.1 --serialized-file squeezenet1_1-jit.pt --handler image_classifier --export-path /home/model-server/model-store
35+
torch-model-archiver --model-name squeezenet1_1 --version 1.1 --serialized-file squeezenet1_1-jit.pt --handler image_classifier --export-path /home/model-server/model-store --force
3636
rm -rf squeezenet1_1-jit.pt
3737
else
3838
echo "Only cpu and xpu devices supported"

0 commit comments

Comments
 (0)