altawil commited on
Commit
afdf05f
·
verified ·
1 Parent(s): 8e923f7

Update model_definition.py

Browse files
Files changed (1) hide show
  1. model_definition.py +1 -1
model_definition.py CHANGED
@@ -1254,7 +1254,7 @@ def load_and_prepare_model(device: torch.device) -> Interfuser:
1254
  config = get_master_config()
1255
 
1256
  # 2. إنشاء النموذج باستخدام إعدادات النموذج فقط
1257
- model = Interfuser(**config['model_params']).to(device)
1258
  logging.info(f"Model instantiated on device: {device}")
1259
 
1260
  # 3. تحميل الأوزان باستخدام الدالة الداخلية للنموذج
 
1254
  config = get_master_config()
1255
 
1256
  # 2. إنشاء النموذج باستخدام إعدادات النموذج فقط
1257
+ model = InterfuserModel(**config['model_params']).to(device)
1258
  logging.info(f"Model instantiated on device: {device}")
1259
 
1260
  # 3. تحميل الأوزان باستخدام الدالة الداخلية للنموذج