BaseerAI commited on
Commit
d24a912
·
verified ·
1 Parent(s): 03d16c5

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():
1243
  # --- القسم 1: معلومات مستودع النموذج على Hugging Face Hub ---
1244
  huggingface_repo = {
1245
  'repo_id': "BaseerAI/Interfuser-Baseer-v1", # استبدله باسم مستودع النموذج الخاص بك
1246
- 'filename': "interfuser_hdpe_v1.pth" # اسم ملف الأوزان داخل المستودع
1247
  }
1248
 
1249
  # --- القسم 2: إعدادات بنية نموذج Interfuser ---
 
1243
  # --- القسم 1: معلومات مستودع النموذج على Hugging Face Hub ---
1244
  huggingface_repo = {
1245
  'repo_id': "BaseerAI/Interfuser-Baseer-v1", # استبدله باسم مستودع النموذج الخاص بك
1246
+ 'filename': "pytorch_model.bin" # اسم ملف الأوزان داخل المستودع
1247
  }
1248
 
1249
  # --- القسم 2: إعدادات بنية نموذج Interfuser ---