vincenzocivale
commited on
Commit
·
3f4d178
1
Parent(s):
96c02e7
Add: implement class registration for scBloodClassifier
Browse files- __init__.py +0 -0
- 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)
|