HuangLab
/

CELL-E_2_OpenCell_2560

Model card Files Files and versions

Emaad commited on May 15, 2023

Commit

e896083

·

1 Parent(s): 9fa6a74

Update config.yaml

Files changed (1) hide show

config.yaml +2 -2

config.yaml CHANGED Viewed

@@ -1,7 +1,7 @@
 model:
   target: celle_main.CELLE_trainer
   params:
-    ckpt_path:
     condition_model_path:
     condition_config_path: nucleus_vqgan.yaml
     vqgan_model_path:
@@ -21,4 +21,4 @@ model:
     fixed_embedding: true
     text_embedding: esm2
     loss_img_weight: 1
-    loss_cond_weight: 1

 model:
   target: celle_main.CELLE_trainer
   params:
+    ckpt_path: model.ckpt
     condition_model_path:
     condition_config_path: nucleus_vqgan.yaml
     vqgan_model_path:
     fixed_embedding: true
     text_embedding: esm2
     loss_img_weight: 1
+    loss_cond_weight: 1