This commit is contained in:
Yousef Rafat 2026-05-14 20:30:14 +03:00
parent efc5141fb0
commit 0743ffb1be
2 changed files with 1 additions and 1 deletions

View File

@ -6,7 +6,7 @@ import torch.nn.functional as F
from fractions import Fraction from fractions import Fraction
from dataclasses import dataclass from dataclasses import dataclass
from typing import List, Any, Dict, Optional, overload, Union, Tuple from typing import List, Any, Dict, Optional, overload, Union, Tuple
from comfy.ldm.trellis2.cumesh import TorchHashMap, Mesh, sparse_submanifold_conv3d from comfy.ldm.trellis2.flexgemm import TorchHashMap, Mesh, sparse_submanifold_conv3d
def pixel_shuffle_3d(x: torch.Tensor, scale_factor: int) -> torch.Tensor: def pixel_shuffle_3d(x: torch.Tensor, scale_factor: int) -> torch.Tensor: