PKUWilliamYang commited on
Commit
5d9f15f
1 Parent(s): d6e1852

Update webUI/styleganex_model.py

Browse files
Files changed (1) hide show
  1. webUI/styleganex_model.py +1 -1
webUI/styleganex_model.py CHANGED
@@ -358,7 +358,7 @@ class Model():
358
  y_hat = self.pspex(x1=x1, x2=x2, use_skip=self.pspex.opts.use_skip, zero_noise=True, resize=False)
359
  y_hat = torch.clamp(y_hat, -1, 1)
360
 
361
- return self.tensor2np(y_hat[0]), 'Done!'
362
 
363
 
364
  def process_vtoonify(self, input_video: str, style_type: str, frame_num: int) -> tuple[list[np.ndarray], str]:
 
358
  y_hat = self.pspex(x1=x1, x2=x2, use_skip=self.pspex.opts.use_skip, zero_noise=True, resize=False)
359
  y_hat = torch.clamp(y_hat, -1, 1)
360
 
361
+ return self.tensor2np(y_hat[0])
362
 
363
 
364
  def process_vtoonify(self, input_video: str, style_type: str, frame_num: int) -> tuple[list[np.ndarray], str]: