This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
modelscope
/
ModelScope
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
0
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
[to
#42322933
] fix bug of CPU running error for multimodal model GEMM
Link:
https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9756849
master
lingchen.zlm
yingda.chen
3 years ago
parent
dac2167b7a
commit
c5a06c259c
1 changed files
with
1 additions
and
0 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-0
modelscope/models/multi_modal/gemm/gemm_model.py
+ 1
- 0
modelscope/models/multi_modal/gemm/gemm_model.py
View File
@@ -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),
Write
Preview
Loading…
Cancel
Save