altawil commited on
Commit
49f417f
·
verified ·
1 Parent(s): afdf05f

Update model_definition.py

Browse files
Files changed (1) hide show
  1. model_definition.py +1 -1
model_definition.py CHANGED
@@ -1243,7 +1243,7 @@ def get_master_config(model_path="model/best_model.pth"):
1243
  return master_config
1244
 
1245
 
1246
- def load_and_prepare_model(device: torch.device) -> Interfuser:
1247
  """
1248
  [النسخة النهائية الصحيحة - تستقبل مدخلاً واحدًا فقط]
1249
  تستخدم دالة الإعدادات الرئيسية لإنشاء وتحميل النموذج.
 
1243
  return master_config
1244
 
1245
 
1246
+ def load_and_prepare_model(device: torch.device) -> InterfuserModel:
1247
  """
1248
  [النسخة النهائية الصحيحة - تستقبل مدخلاً واحدًا فقط]
1249
  تستخدم دالة الإعدادات الرئيسية لإنشاء وتحميل النموذج.