Instructions to use ctheodoris/Geneformer with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use ctheodoris/Geneformer with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="ctheodoris/Geneformer")# Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("ctheodoris/Geneformer") model = AutoModelForMaskedLM.from_pretrained("ctheodoris/Geneformer", device_map="auto") - Inference
- Notebooks
- Google Colab
- Kaggle
Christina Theodoris commited on
Commit ·
f17618a
1
Parent(s): f953554
update collator with dummy save fn for transformers compatibility
Browse files
geneformer/collator_for_classification.py
CHANGED
|
@@ -103,6 +103,9 @@ class PrecollatorForGeneAndCellClassification(SpecialTokensMixin):
|
|
| 103 |
def pad_token_id(self):
|
| 104 |
return self._pad_token_id
|
| 105 |
|
|
|
|
|
|
|
|
|
|
| 106 |
def _get_padding_truncation_strategies(
|
| 107 |
self,
|
| 108 |
padding=True,
|
|
|
|
| 103 |
def pad_token_id(self):
|
| 104 |
return self._pad_token_id
|
| 105 |
|
| 106 |
+
def save_pretrained(self, save_directory):
|
| 107 |
+
pass
|
| 108 |
+
|
| 109 |
def _get_padding_truncation_strategies(
|
| 110 |
self,
|
| 111 |
padding=True,
|