File size: 373 Bytes
3f4d178 b4ea423 3f4d178 |
1 2 3 4 5 6 7 8 9 |
# 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) |