scBloodClassifier / class_registration.py
vincenzocivale
Refactor: update import paths and model type for scBloodClassifier; remove unified model implementation
b4ea423
# Import the custom classes
from .modeling_scBloodClassifier import scBloodClassifierConfig, scBloodClassifier
# Import the necessary Auto classes from transformers
from transformers import AutoConfig, AutoModel
# Register your custom classes
AutoConfig.register("scBloodClassifier", scBloodClassifierConfig)
AutoModel.register(scBloodClassifierConfig, scBloodClassifier)