Ole-Christian Galbo Engstrøm
commited on
Commit
·
cee5707
1
Parent(s):
becf17e
Cleanup
Browse files- __init__.py +0 -10
__init__.py
DELETED
|
@@ -1,10 +0,0 @@
|
|
| 1 |
-
from transformers import AutoConfig, AutoModel
|
| 2 |
-
|
| 3 |
-
from .unet_config import UNetConfig
|
| 4 |
-
from .unet_hf import UNetModel
|
| 5 |
-
|
| 6 |
-
# Register with Hugging Face Auto classes
|
| 7 |
-
AutoConfig.register("unet", UNetConfig)
|
| 8 |
-
AutoModel.register(UNetConfig, UNetModel)
|
| 9 |
-
|
| 10 |
-
__all__ = ["UNetConfig", "UNetModel"]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|