macwiatrak
/

bacformer-masked-complete-genomes

Model card Files Files and versions Community

macwiatrak commited on May 27

Commit

ff905f4

·

verified ·

1 Parent(s): fbaa9b1

Update config.json

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -8,7 +8,8 @@
     "AutoConfig": "configuration_bacformer.BacformerConfig",
     "AutoModel": "modeling_bacformer.BacformerModel",
     "AutoModelForMaskedLM": "modeling_bacformer.BacformerForMaskedGM",
-    "AutoModelForTokenClassification": "modeling_bacformer.BacformerForProteinClassification"
   },
   "batch_size": 1,
   "ckpt_path": null,

     "AutoConfig": "configuration_bacformer.BacformerConfig",
     "AutoModel": "modeling_bacformer.BacformerModel",
     "AutoModelForMaskedLM": "modeling_bacformer.BacformerForMaskedGM",
+    "AutoModelForTokenClassification": "modeling_bacformer.BacformerForProteinClassification",
+    "AutoModelForSequenceClassification": "modeling_bacformer.BacformerForGenomeClassification"
   },
   "batch_size": 1,
   "ckpt_path": null,