Update handler.py
Browse files- handler.py +2 -1
handler.py
CHANGED
@@ -37,7 +37,8 @@ class EndpointHandler:
|
|
37 |
"dalle": 2,
|
38 |
"real": 3,
|
39 |
"other_ai": 4
|
40 |
-
}
|
|
|
41 |
)
|
42 |
|
43 |
# محاولة تحميل الأوزان المخصصة إذا كانت موجودة
|
|
|
37 |
"dalle": 2,
|
38 |
"real": 3,
|
39 |
"other_ai": 4
|
40 |
+
},
|
41 |
+
ignore_mismatched_sizes=True # إضافة هذه المعلمة لتجاهل عدم تطابق الأحجام
|
42 |
)
|
43 |
|
44 |
# محاولة تحميل الأوزان المخصصة إذا كانت موجودة
|