Spaces:
Running
Running
altawil
commited on
Update model_definition.py
Browse files- 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 =
|
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. تحميل الأوزان باستخدام الدالة الداخلية للنموذج
|