alibabasglab commited on
Commit
4aa542b
·
verified ·
1 Parent(s): 663300b

Update extract_everything.py

Browse files
Files changed (1) hide show
  1. extract_everything.py +1 -1
extract_everything.py CHANGED
@@ -41,7 +41,7 @@ class main(nn.Module):
41
 
42
 
43
  class extract_everything:
44
- def __init__(self, model_name):
45
  self.args_wrapper = main()
46
  self.model = self.args_wrapper(model_name)
47
 
 
41
 
42
 
43
  class extract_everything:
44
+ def __init__(self, model_name="EEYD_locoformer"):
45
  self.args_wrapper = main()
46
  self.model = self.args_wrapper(model_name)
47