Update paths

README.md

@@ -2,7 +2,7 @@
 datasets:
 - Bo1015/fluorescence_prediction
 base_model:
-- genbio-ai/proteinMoE-16b
+- genbio-ai/AIDO.Protein-16B
 ---
 LoRA fine-tuned checkpoint for fluorescence_prediction.
 
@@ -12,7 +12,7 @@ LoRA fine-tuned checkpoint for fluorescence_prediction.
 from huggingface_hub import snapshot_download
 from pathlib import Path
 
-model_name = "genbio-ai/proteinfm-16b-fluorescence_prediction-ckpt"
+model_name = "genbio-ai/AIDO.Protein-16B-fluorescence_prediction"
 genbio_models_path = Path.home().joinpath('genbio_models', model_name)
 genbio_models_path.mkdir(parents=True, exist_ok=True)
 snapshot_download(repo_id=model_name, local_dir=genbio_models_path)

config.yaml

@@ -36,14 +36,14 @@ trainer:
   logger:
     class_path: lightning.pytorch.loggers.WandbLogger
     init_args:
-      name: fluorescence_proteinmoe_16B
+      name: fluorescence_AIDO.Protein-16B
       save_dir: lightning_logs/protein
       version: null
       offline: false
       dir: null
       id: null
       anonymous: null
-      project: xtrimo_benchmark_gbft_reproduction
+      project: xtrimo_benchmark_mgen_reproduction
       log_model: false
       experiment: null
       prefix: ''
@@ -77,7 +77,7 @@ trainer:
       log_weight_decay: false
   - class_path: lightning.pytorch.callbacks.ModelCheckpoint
     init_args:
-      dirpath: modelgenerator/logs/xtrimo_benchmark_gbft_reproduction/fluorescence_proteinmoe_16B
+      dirpath: logs/xtrimo_benchmark_mgen_reproduction/fluorescence_AIDO.Protein-16B
       filename: best_val:{epoch}-{val_spearman:.3f}
       monitor: val_spearman
       verbose: false
@@ -141,7 +141,7 @@ model:
   class_path: modelgenerator.tasks.SequenceRegression
   init_args:
     backbone:
-      class_path: modelgenerator.backbones.proteinfm
+      class_path: modelgenerator.backbones.aido_protein_16b
       init_args:
         from_scratch: false
         max_length: 512
@@ -202,7 +202,7 @@ model:
 data:
   class_path: modelgenerator.data.FluorescencePrediction
   init_args:
-    path: biomap-research/fluorescence_prediction
+    path: proteinglm/fluorescence_prediction
     normalize: true
     config_name: null
     train_split_name: train