Method to

wan/modules/animate/animate_utils.py:36–37

(self, *args, **kwargs)

        self.tensors = tensors
    
    def to(self, *args, **kwargs):
        return TensorList([u.to(*args, **kwargs) for u in self.tensors])
    
    def size(self, dim):
        assert dim == 0, 'only support get the 0th size'