add explicit chemical encoder params

3b-model/model_config.yml

@@ -37,9 +37,8 @@ expression_decoder:
   n_layers: 1
   activation: gelu
 chemical_encoder:
-  drug_fps_path:
-    remote: s3://vevo-ml-datasets/mosaicfm_v2/datasets/drug_fps_pad.npy
-    local: drug_fps_pad.npy
+  num_drugs: 378
+  fp_dim: 2048
   activation: gelu
   padding_idx: 0
   freeze: false