From c5a06c259c5aac6ee6397ef60a9610f48714ddb9 Mon Sep 17 00:00:00 2001
From: "lingchen.zlm" <lingchen.zlm@alibaba-inc.com>
Date: Mon, 15 Aug 2022 16:32:36 +0800
Subject: [PATCH] [to #42322933] fix bug of CPU running error for multimodal
 model GEMM         Link:
 https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9756849

---
 modelscope/models/multi_modal/gemm/gemm_model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/modelscope/models/multi_modal/gemm/gemm_model.py b/modelscope/models/multi_modal/gemm/gemm_model.py
index a5380858..356dc8d3 100644
--- a/modelscope/models/multi_modal/gemm/gemm_model.py
+++ b/modelscope/models/multi_modal/gemm/gemm_model.py
@@ -45,6 +45,7 @@ class GEMMForMultiModalEmbedding(TorchModel):
             self.gemm_model.to('cuda:{}'.format(self.device_id))
             logger.info('Use GPU: {}'.format(self.device_id))
         else:
+            self.device_id = -1
             logger.info('Use CPU for inference')
         self.img_preprocessor = T.Compose([
             T.Resize(224),