Spaces:
Runtime error
Runtime error
Update stf_utils.py
Browse files- stf_utils.py +1 -1
stf_utils.py
CHANGED
|
@@ -106,7 +106,7 @@ class STFPipeline:
|
|
| 106 |
|
| 107 |
|
| 108 |
self.config_path = os.path.join(stf_path, config_path)
|
| 109 |
-
self.checkpoint_path = os.path.join(stf_path, checkpoint_path)
|
| 110 |
#self.work_root_path = os.path.join(stf_path, root_path)
|
| 111 |
self.work_root_path = os.path.join(root_path)
|
| 112 |
self.device = device
|
|
|
|
| 106 |
|
| 107 |
|
| 108 |
self.config_path = os.path.join(stf_path, config_path)
|
| 109 |
+
self.checkpoint_path = os.path.join('/tmp/stf/', checkpoint_path) #stf_path, checkpoint_path)
|
| 110 |
#self.work_root_path = os.path.join(stf_path, root_path)
|
| 111 |
self.work_root_path = os.path.join(root_path)
|
| 112 |
self.device = device
|