sentence-transformers-testing
/

stsb-bert-tiny-lora

Sentence Similarity

sentence-transformers

feature-extraction

Generated from Trainer

dataset_size:3012496

loss:MultipleNegativesRankingLoss

Inference Endpoints

Model card Files Files and versions Community

tomaarsen HF staff commited on 13 days ago

Commit

3b4f75b

•

1 Parent(s): b30edbd

Update train_script.py

Files changed (1) hide show

train_script.py +1 -1

train_script.py CHANGED Viewed

@@ -25,7 +25,7 @@ model = SentenceTransformer(
     ),
 )
-# Apply PEFT with PromptTuningConfig
 peft_config = LoraConfig(
     task_type=TaskType.FEATURE_EXTRACTION,
     inference_mode=False,

     ),
 )
+# Apply a PEFT Adapter
 peft_config = LoraConfig(
     task_type=TaskType.FEATURE_EXTRACTION,
     inference_mode=False,