Skip to content

Commit 1252b45

Browse files
committed
erge branch 'gliner' of https://github.com/HumanSignal/label-studio-ml-backend into gliner
2 parents 1f5970e + 24b500a commit 1252b45

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

label_studio_ml/examples/gliner/model.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -189,6 +189,7 @@ def train(self, model, training_args, train_data, eval_data=None):
189189
logger.info(f"Model Trained, saving to {ckpt} ")
190190
trainer.save_model(ckpt)
191191

192+
192193
def fit(self, event, data, **kwargs):
193194
"""
194195
This method is called each time an annotation is created or updated
@@ -234,6 +235,7 @@ def fit(self, event, data, **kwargs):
234235

235236
training_args = TrainingArguments(
236237
output_dir="models/training_output",
238+
237239
learning_rate=5e-6,
238240
weight_decay=0.01,
239241
others_lr=1e-5,

label_studio_ml/examples/watsonx_llm/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,4 @@ pydantic>=1.10.0
44
langchain==0.1.8
55
langchain_ibm==0.1.0
66
pytesseract==0.3.10
7-
label_studio_sdk==1.0.4
7+
label_studio_sdk==1.0.10

0 commit comments

Comments
 (0)