Update hf_model.py
Browse files- hf_model.py +1 -0
hf_model.py
CHANGED
|
@@ -39,6 +39,7 @@ from .vit_patch_generator import ViTPatchGenerator
|
|
| 39 |
from .vitdet import apply_vitdet_arch, VitDetArgs
|
| 40 |
|
| 41 |
# Register extra models
|
|
|
|
| 42 |
from .extra_timm_models import *
|
| 43 |
|
| 44 |
|
|
|
|
| 39 |
from .vitdet import apply_vitdet_arch, VitDetArgs
|
| 40 |
|
| 41 |
# Register extra models
|
| 42 |
+
from . import dinov2_arch
|
| 43 |
from .extra_timm_models import *
|
| 44 |
|
| 45 |
|