Fix AutoModel registration by adding dynamic registration code
Browse files- modeling_interfuser.py +1 -1
modeling_interfuser.py
CHANGED
|
@@ -107,7 +107,7 @@ def resnet50d(**kwargs):
|
|
| 107 |
class HybridEmbed(nn.Module):
|
| 108 |
def __init__(self, backbone, img_size=224, patch_size=1, feature_size=None, in_chans=3, embed_dim=768):
|
| 109 |
super().__init__()
|
| 110 |
-
|
| 111 |
# --- بداية التعديلات ---
|
| 112 |
# تعديل 1: تأكد من أن img_size هو tuple للوصول الآمن إلى عناصره
|
| 113 |
if isinstance(img_size, int):
|
|
|
|
| 107 |
class HybridEmbed(nn.Module):
|
| 108 |
def __init__(self, backbone, img_size=224, patch_size=1, feature_size=None, in_chans=3, embed_dim=768):
|
| 109 |
super().__init__()
|
| 110 |
+
print("--> Initializing HybridEmbed...") # <--- أضف هذا
|
| 111 |
# --- بداية التعديلات ---
|
| 112 |
# تعديل 1: تأكد من أن img_size هو tuple للوصول الآمن إلى عناصره
|
| 113 |
if isinstance(img_size, int):
|