videos: A list of videos each with shape [C, T, H, W].
(self, videos)
| 645 | ).eval().requires_grad_(False).to(device) |
| 646 | |
| 647 | def encode(self, videos): |
| 648 | """ |
| 649 | videos: A list of videos each with shape [C, T, H, W]. |
| 650 | """ |
| 651 | with amp.autocast(dtype=self.dtype): |
| 652 | return [ |
| 653 | self.model.encode(u.unsqueeze(0), self.scale).float().squeeze(0) |
| 654 | for u in videos |
| 655 | ] |
| 656 | |
| 657 | def decode(self, zs): |
| 658 | with amp.autocast(dtype=self.dtype): |
no outgoing calls
no test coverage detected