igoramf commited on
Commit
c9b1cb7
·
verified ·
1 Parent(s): dd94905

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -0
README.md CHANGED
@@ -25,3 +25,18 @@ target_modules = ['q_lin']
25
 
26
  ## F1_SCORE = 0.914643
27
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25
 
26
  ## F1_SCORE = 0.914643
27
 
28
+
29
+ ## HOW TO USE
30
+ model_repo = "igoramf/lora-pt-sentiment-analysis"
31
+
32
+ config = PeftConfig.from_pretrained(model_repo)
33
+ model = AutoModelForSequenceClassification.from_pretrained(
34
+ config.base_model_name_or_path,
35
+ num_labels=3
36
+ )
37
+
38
+ tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
39
+ tokenizer.pad_token = tokenizer.eos_token
40
+
41
+ model = PeftModel.from_pretrained(model, model_repo)
42
+