Upload model
Browse files- eradio_model.py +2 -0
eradio_model.py
CHANGED
|
@@ -24,6 +24,8 @@ import numpy as np
|
|
| 24 |
import torch.nn.functional as F
|
| 25 |
import warnings
|
| 26 |
|
|
|
|
|
|
|
| 27 |
|
| 28 |
SIMPLER_UP_TOWER = False
|
| 29 |
|
|
|
|
| 24 |
import torch.nn.functional as F
|
| 25 |
import warnings
|
| 26 |
|
| 27 |
+
# Register extra models
|
| 28 |
+
from . import extra_timm_models
|
| 29 |
|
| 30 |
SIMPLER_UP_TOWER = False
|
| 31 |
|