toind commited on
Commit
dc8ca3d
·
verified ·
1 Parent(s): 558b35e

Update __init__.py

Browse files
Files changed (1) hide show
  1. __init__.py +3 -1
__init__.py CHANGED
@@ -1,5 +1,7 @@
 
 
1
  from transformers import AutoConfig, AutoModel
2
- from .custom_model import CustomConfig, CustomModel
3
 
4
  # Register the custom classes
5
  AutoConfig.register("custom_model", CustomConfig)
 
1
+ # __init__.py
2
+
3
  from transformers import AutoConfig, AutoModel
4
+ from .custom_model2 import CustomConfig, CustomModel
5
 
6
  # Register the custom classes
7
  AutoConfig.register("custom_model", CustomConfig)