Caleb Ellington commited on
Commit
9625014
·
1 Parent(s): beaed13

Update paths

Browse files
Files changed (2) hide show
  1. README.md +2 -2
  2. config.yaml +5 -5
README.md CHANGED
@@ -2,7 +2,7 @@
2
  datasets:
3
  - Bo1015/ssp_q3
4
  base_model:
5
- - genbio-ai/proteinMoE-16b
6
  ---
7
  LoRA fine-tuned checkpoint for ssp_q3.
8
 
@@ -12,7 +12,7 @@ LoRA fine-tuned checkpoint for ssp_q3.
12
  from huggingface_hub import snapshot_download
13
  from pathlib import Path
14
 
15
- model_name = "genbio-ai/proteinfm-16b-ssp_q3-ckpt"
16
  genbio_models_path = Path.home().joinpath('genbio_models', model_name)
17
  genbio_models_path.mkdir(parents=True, exist_ok=True)
18
  snapshot_download(repo_id=model_name, local_dir=genbio_models_path)
 
2
  datasets:
3
  - Bo1015/ssp_q3
4
  base_model:
5
+ - genbio-ai/AIDO.Protein-16B
6
  ---
7
  LoRA fine-tuned checkpoint for ssp_q3.
8
 
 
12
  from huggingface_hub import snapshot_download
13
  from pathlib import Path
14
 
15
+ model_name = "genbio-ai/AIDO.Protein-16B-ssp_q3"
16
  genbio_models_path = Path.home().joinpath('genbio_models', model_name)
17
  genbio_models_path.mkdir(parents=True, exist_ok=True)
18
  snapshot_download(repo_id=model_name, local_dir=genbio_models_path)
config.yaml CHANGED
@@ -36,14 +36,14 @@ trainer:
36
  logger:
37
  class_path: lightning.pytorch.loggers.WandbLogger
38
  init_args:
39
- name: ssp_proteinmoe_16B
40
  save_dir: lightning_logs/protein
41
  version: null
42
  offline: false
43
  dir: null
44
  id: null
45
  anonymous: null
46
- project: xtrimo_benchmark_gbft_reproduction
47
  log_model: false
48
  experiment: null
49
  prefix: ''
@@ -77,7 +77,7 @@ trainer:
77
  log_weight_decay: false
78
  - class_path: lightning.pytorch.callbacks.ModelCheckpoint
79
  init_args:
80
- dirpath: modelgenerator/logs/xtrimo_benchmark_gbft_reproduction/ssp_proteinmoe_16B
81
  filename: best_val:{epoch}-{val_accuracy:.3f}
82
  monitor: val_accuracy
83
  verbose: false
@@ -149,7 +149,7 @@ model:
149
  dropout: 0.1
150
  dropout_in_middle: true
151
  backbone:
152
- class_path: modelgenerator.backbones.proteinfm
153
  init_args:
154
  from_scratch: false
155
  max_length: 512
@@ -199,7 +199,7 @@ model:
199
  data:
200
  class_path: modelgenerator.data.SspQ3
201
  init_args:
202
- path: biomap-research/ssp_q3
203
  batch_size: 1
204
  max_length: 512
205
  config_name: null
 
36
  logger:
37
  class_path: lightning.pytorch.loggers.WandbLogger
38
  init_args:
39
+ name: ssp_AIDO.Protein-16B
40
  save_dir: lightning_logs/protein
41
  version: null
42
  offline: false
43
  dir: null
44
  id: null
45
  anonymous: null
46
+ project: xtrimo_benchmark_mgen_reproduction
47
  log_model: false
48
  experiment: null
49
  prefix: ''
 
77
  log_weight_decay: false
78
  - class_path: lightning.pytorch.callbacks.ModelCheckpoint
79
  init_args:
80
+ dirpath: logs/xtrimo_benchmark_mgen_reproduction/ssp_AIDO.Protein-16B
81
  filename: best_val:{epoch}-{val_accuracy:.3f}
82
  monitor: val_accuracy
83
  verbose: false
 
149
  dropout: 0.1
150
  dropout_in_middle: true
151
  backbone:
152
+ class_path: modelgenerator.backbones.aido_protein_16b
153
  init_args:
154
  from_scratch: false
155
  max_length: 512
 
199
  data:
200
  class_path: modelgenerator.data.SspQ3
201
  init_args:
202
+ path: proteinglm/ssp_q3
203
  batch_size: 1
204
  max_length: 512
205
  config_name: null