metadata
tags:
- model_hub_mixin
- pytorch_model_hub_mixin
How to use
Until its next release, the transformers library needs to be installed from source with the following command in order to use the models. PyTorch should also be installed.
pip install --upgrade git+https://github.com/huggingface/transformers.git
pip install torch
A small snippet of code is given here in order to infer with the model from random input.
import torch
from transformers import AutoModel
model = AutoModel.from_pretrained(
"InstaDeepAI/sCellTransformer",
trust_remote_code=True,
)
num_cells = model.config.num_cells
dummy_gene_expressions = torch.randint(0, 5, (1, 19968 * num_cells))
torch_output = model(dummy_gene_expressions)