# 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)