mohammed-aljafry commited on
Commit
cf25dfa
·
verified ·
1 Parent(s): b80f4a3

Fix AutoModel registration by adding dynamic registration code

Browse files
Files changed (1) hide show
  1. 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):