vincenzocivale commited on
Commit
3f4d178
·
1 Parent(s): 96c02e7

Add: implement class registration for scBloodClassifier

Browse files
Files changed (2) hide show
  1. __init__.py +0 -0
  2. class_registration.py +9 -0
__init__.py ADDED
File without changes
class_registration.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ # Import the custom classes
2
+ from .modeling_scBloodClassifier import scBloodClassifierConfig, scBloodClassifier
3
+
4
+ # Import the necessary Auto classes from transformers
5
+ from transformers import AutoConfig, AutoModel
6
+
7
+ # Register your custom classes
8
+ AutoConfig.register("scBloodClassifier", scBloodClassifierConfig)
9
+ AutoModel.register(scBloodClassifierConfig, scBloodClassifier)