yerang commited on
Commit
c05f460
·
verified ·
1 Parent(s): 37beeb5

Update stf_utils.py

Browse files
Files changed (1) hide show
  1. stf_utils.py +2 -1
stf_utils.py CHANGED
@@ -86,7 +86,8 @@ class STFPipeline:
86
 
87
  self.config_path = os.path.join(stf_path, config_path)
88
  self.checkpoint_path = os.path.join(stf_path, checkpoint_path)
89
- self.work_root_path = os.path.join(stf_path, root_path)
 
90
  self.device = device
91
  self.template_video_path=os.path.join(stf_path, template_video_path)
92
 
 
86
 
87
  self.config_path = os.path.join(stf_path, config_path)
88
  self.checkpoint_path = os.path.join(stf_path, checkpoint_path)
89
+ #self.work_root_path = os.path.join(stf_path, root_path)
90
+ self.work_root_path = os.path.join(root_path)
91
  self.device = device
92
  self.template_video_path=os.path.join(stf_path, template_video_path)
93