docker build 构建修改
This commit is contained in:
19
rag/res/deepdoc/README.md
Normal file
19
rag/res/deepdoc/README.md
Normal file
@@ -0,0 +1,19 @@
|
||||
---
|
||||
license: apache-2.0
|
||||
---
|
||||
|
||||
### Model Loading
|
||||
```python
|
||||
import xgboost as xgb
|
||||
import torch
|
||||
|
||||
model = xgb.Booster()
|
||||
if torch.cuda.is_available():
|
||||
model.set_param({"device": "cuda"})
|
||||
model.load_model('InfiniFlow/text_concat_xgb_v1.0')
|
||||
```
|
||||
|
||||
### Prediction
|
||||
```python
|
||||
model.predict(xgb.DMatrix([feature]))[0]
|
||||
```
|
||||
Reference in New Issue
Block a user