mirror of
https://github.com/FlagOpen/FlagEmbedding.git
synced 2025-06-27 02:39:58 +00:00
use logger
This commit is contained in:
parent
97f57a1b92
commit
95ab52eb9f
@ -82,10 +82,10 @@ class BGEM3Model(nn.Module):
|
||||
|
||||
if os.path.exists(os.path.join(model_name, 'colbert_linear.pt')) and os.path.exists(
|
||||
os.path.join(model_name, 'sparse_linear.pt')):
|
||||
print('loading existing colbert_linear and sparse_linear---------')
|
||||
logger.info('loading existing colbert_linear and sparse_linear---------')
|
||||
self.load_pooler(model_dir=model_name)
|
||||
else:
|
||||
print(
|
||||
logger.info(
|
||||
'The parameters of colbert_linear and sparse linear is new initialize. Make sure the model is loaded for training, not inferencing')
|
||||
|
||||
def gradient_checkpointing_enable(self, **kwargs):
|
||||
|
@ -15,14 +15,10 @@ pip install -U FlagEmbedding
|
||||
```
|
||||
git clone https://github.com/FlagOpen/FlagEmbedding.git
|
||||
cd FlagEmbedding
|
||||
pip install .
|
||||
```
|
||||
For development, install as editable:
|
||||
|
||||
```
|
||||
pip install -e .
|
||||
```
|
||||
|
||||
|
||||
## 2. Data format
|
||||
|
||||
Training data should be a jsonl file, where each line is a dict like this:
|
||||
|
Loading…
x
Reference in New Issue
Block a user