AmelieSchreiber commited on
Commit
bab67c3
1 Parent(s): b20b374

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -40,7 +40,7 @@ import torch
40
  # Path to the saved LoRA model
41
  model_path = "AmelieSchreiber/esm2_t12_35M_LoRA_RNA_binding"
42
  # ESM2 base model
43
- base_model_path = "facebook/esm2_t6_8M_UR50D"
44
 
45
  # Load the model
46
  base_model = AutoModelForTokenClassification.from_pretrained(base_model_path)
 
40
  # Path to the saved LoRA model
41
  model_path = "AmelieSchreiber/esm2_t12_35M_LoRA_RNA_binding"
42
  # ESM2 base model
43
+ base_model_path = "facebook/esm2_t12_35M_UR50D"
44
 
45
  # Load the model
46
  base_model = AutoModelForTokenClassification.from_pretrained(base_model_path)