AmelieSchreiber commited on
Commit
3975a11
1 Parent(s): 2839395

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -14,7 +14,7 @@ import torch
14
  # Path to the saved LoRA model
15
  model_path = "AmelieSchreiber/esm2_t33_650M_qlora_binding_12M"
16
  # ESM2 base model
17
- base_model_path = "facebook/esm2_t12_35M_UR50D"
18
 
19
  # Load the model
20
  base_model = AutoModelForTokenClassification.from_pretrained(base_model_path)
 
14
  # Path to the saved LoRA model
15
  model_path = "AmelieSchreiber/esm2_t33_650M_qlora_binding_12M"
16
  # ESM2 base model
17
+ base_model_path = "facebook/esm2_t33_650M_UR50D"
18
 
19
  # Load the model
20
  base_model = AutoModelForTokenClassification.from_pretrained(base_model_path)