docker build 构建修改

This commit is contained in:
2025-11-06 17:15:46 +08:00
parent 33e09afc54
commit ecdb4321e8
58 changed files with 72216 additions and 237 deletions

19
rag/res/deepdoc/README.md Normal file
View 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]
```