Spaces:
Sleeping
Sleeping
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. تحميل الأوزان باستخدام الدالة الداخلية للنموذج
|