jbilcke-hf HF staff commited on
Commit
a54c820
1 Parent(s): 829c2a5

Update handler.py

Browse files
Files changed (1) hide show
  1. 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=float, default=1.0)
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