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:
|