Spaces:
Sleeping
Sleeping
altawil
commited on
Update model_definition.py
Browse files- 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) ->
|
| 1247 |
"""
|
| 1248 |
[النسخة النهائية الصحيحة - تستقبل مدخلاً واحدًا فقط]
|
| 1249 |
تستخدم دالة الإعدادات الرئيسية لإنشاء وتحميل النموذج.
|
|
|
|
| 1243 |
return master_config
|
| 1244 |
|
| 1245 |
|
| 1246 |
+
def load_and_prepare_model(device: torch.device) -> InterfuserModel:
|
| 1247 |
"""
|
| 1248 |
[النسخة النهائية الصحيحة - تستقبل مدخلاً واحدًا فقط]
|
| 1249 |
تستخدم دالة الإعدادات الرئيسية لإنشاء وتحميل النموذج.
|