Spaces:
Sleeping
Sleeping
PKUWilliamYang
commited on
Commit
•
5d9f15f
1
Parent(s):
d6e1852
Update webUI/styleganex_model.py
Browse files
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])
|
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]:
|