Spaces:
Running
on
Zero
Running
on
Zero
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
|