yerang commited on
Commit
b318e15
1 Parent(s): 363d8a1

Update stf_utils.py

Browse files
Files changed (1) hide show
  1. 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)
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