Commit
•
a54c820
1
Parent(s):
829c2a5
Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -56,7 +56,7 @@ def get_default_args():
|
|
56 |
|
57 |
# Model paths and weights
|
58 |
parser.add_argument("--model-base", type=str, default=None)
|
59 |
-
parser.add_argument("--dit-weight", type=
|
60 |
parser.add_argument("--load-key", type=str, default=None)
|
61 |
|
62 |
# Hardware settings
|
|
|
56 |
|
57 |
# Model paths and weights
|
58 |
parser.add_argument("--model-base", type=str, default=None)
|
59 |
+
parser.add_argument("--dit-weight", type=str, default=None)
|
60 |
parser.add_argument("--load-key", type=str, default=None)
|
61 |
|
62 |
# Hardware settings
|