hysts HF staff commited on
Commit
67e1591
1 Parent(s): d074d70
Files changed (1) hide show
  1. model.py +1 -0
model.py CHANGED
@@ -65,6 +65,7 @@ class Model:
65
  self.task_name = ''
66
 
67
  self.model_dir = pathlib.Path(model_dir)
 
68
 
69
  self.use_lightweight = use_lightweight
70
  if use_lightweight:
 
65
  self.task_name = ''
66
 
67
  self.model_dir = pathlib.Path(model_dir)
68
+ self.model_dir.mkdir(exist_ok=True, parents=True)
69
 
70
  self.use_lightweight = use_lightweight
71
  if use_lightweight: