Spaces:
Running
Running
hysts
commited on
Commit
•
385c783
1
Parent(s):
6be65e1
Fix
Browse files
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:
|