Functions — github.com/Stable-X/ComfyUI-Hi3DGen

hub / github.com/Stable-X/ComfyUI-Hi3DGen / functions

Functions333 in github.com/Stable-X/ComfyUI-Hi3DGen

⨍ Functions 333◇ Types & classes 71

Method__init__

( self, in_channels: int, out_channels: int, mode: Literal["conv", "avgpoo

trellis/models/sparse_structure_vae.py:51

Method__init__

( self, in_channels: int, out_channels: int, mode: Literal["conv", "neares

trellis/models/sparse_structure_vae.py:76

Method__init__

( self, in_channels: int, latent_channels: int, num_res_blocks: int,

trellis/models/sparse_structure_vae.py:114

Method__init__

( self, out_channels: int, latent_channels: int, num_res_blocks: int,

trellis/models/sparse_structure_vae.py:223

Method__init__

( self, channels: int, emb_channels: int, out_channels: Optional[int] = No

trellis/models/structured_latent_flow.py:15

Method__init__

( self, in_channels: int, model_channels: int, num_blocks: int, n

trellis/models/structured_latent_vae/base.py:32

Method__init__

( self, channels: int, resolution: int, out_channels: Optional[int] = None

trellis/models/structured_latent_vae/decoder_mesh.py:22

Method__init__

( self, resolution: int, in_channels: int, model_channels: int, l

trellis/models/structured_latent_vae/encoder.py:10

Method__init__

a model to generate a mesh from sparse features structures using flexicube

trellis/representations/mesh/cube2mesh.py:59

Method__init__

(self, device="cuda")

trellis/representations/mesh/flexicube.py:18

Method__init__

( self, vae: AutoencoderKL, unet: UNet2DConditionModel, controlnet: Union[Cont

stablex/pipeline_yoso.py:146

Method__mul__

(self, other: Union[torch.Tensor, 'SparseTensor', float])

trellis/modules/sparse/basic.py:339

Method__neg__

(self)

trellis/modules/sparse/basic.py:309

Method__radd__

(self, other: Union[torch.Tensor, 'SparseTensor', float])

trellis/modules/sparse/basic.py:330

Method__rmul__

(self, other: Union[torch.Tensor, 'SparseTensor', float])

trellis/modules/sparse/basic.py:342

Method__rsub__

(self, other: Union[torch.Tensor, 'SparseTensor', float])

trellis/modules/sparse/basic.py:336

Method__rtruediv__

(self, other: Union[torch.Tensor, 'SparseTensor', float])

trellis/modules/sparse/basic.py:348

Method__sub__

(self, other: Union[torch.Tensor, 'SparseTensor', float])

trellis/modules/sparse/basic.py:333

Method__truediv__

(self, other: Union[torch.Tensor, 'SparseTensor', float])

trellis/modules/sparse/basic.py:345

Method_basic_init

(module)

trellis/models/sparse_structure_flow.py:152

Method_basic_init

(module)

trellis/models/structured_latent_flow.py:210

Method_basic_init

(module)

trellis/models/structured_latent_vae/base.py:103

Method_eps_to_xstart

(self, x_t, t, eps)

trellis/pipelines/samplers/flow_euler.py:24

Method_inference_model

(self, model, x_t, t, cond, neg_cond, cfg_strength, **kwargs)

trellis/pipelines/samplers/classifier_free_guidance_mixin.py:9

Method_inference_model

(self, model, x_t, t, cond, neg_cond, cfg_strength, cfg_interval, **kwargs)

trellis/pipelines/samplers/guidance_interval_mixin.py:9

Method_initialize_transforms

Initialize image transforms if needed.

IF_TrellisCheckpointLoader.py:98

Method_new_inference_model

(self, model, x_t, t, cond, **kwargs)

trellis/pipelines/trellis_image_to_3d.py:453

Method_solve_vd_QEF

(self, p_bxnx3, norm_bxnx3, c_bx3, qef_reg_scale)

trellis/representations/mesh/flexicube.py:207

Method_xstart_to_eps

(self, x_t, t, x_0)

trellis/pipelines/samplers/flow_euler.py:28

Functionatol

Absolute tolerance.

trellis/utils/general_utils.py:164

Methodcleanup

Explicit cleanup method that preserves essential properties

trellis/pipelines/trellis_image_to_3d.py:215

Methodcomput_v_normals

(self, verts, faces)

trellis/representations/mesh/cube2mesh.py:40

Methodconvert_to_fp32

Convert the torso of the model to float32.

trellis/models/sparse_structure_flow.py:144

Methodconvert_to_fp32

Convert the torso of the model to float32.

trellis/models/sparse_structure_vae.py:177

Methodconvert_to_fp32

Convert the torso of the model to float32.

trellis/models/sparse_structure_vae.py:286

Methodconvert_to_fp32

Convert the torso of the model to float32.

trellis/models/structured_latent_flow.py:200

Methodconvert_to_fp32

Convert the torso of the model to float32.

trellis/models/structured_latent_vae/base.py:95

Methodconvert_to_fp32

Convert the torso of the model to float32.

trellis/models/structured_latent_vae/decoder_mesh.py:138

Methodcoords

(self)

trellis/modules/sparse/basic.py:162

Methodcpu

(self)

trellis/modules/sparse/basic.py:214

Methodcuda

(self)

trellis/modules/sparse/basic.py:219

Methoddense

(self)

trellis/modules/sparse/basic.py:237

Methoddevice

(self)

trellis/modules/sparse/basic.py:180

Methoddevice

Override device property to ensure it persists

trellis/pipelines/trellis_image_to_3d.py:209

Methoddevice

Return the device of the model.

trellis/models/sparse_structure_flow.py:132

Methoddevice

Return the device of the model.

trellis/models/sparse_structure_vae.py:162

Methoddevice

Return the device of the model.

trellis/models/sparse_structure_vae.py:271

Methoddevice

Return the device of the model.

trellis/models/structured_latent_flow.py:186

Methoddevice

Return the device of the model.

trellis/models/structured_latent_vae/base.py:83

Functiondict_all

Recursively apply a function to all non-dictionary leaf values in a dictionary.

trellis/utils/general_utils.py:87

Functiondict_any

Recursively apply a function to all non-dictionary leaf values in a dictionary.

trellis/utils/general_utils.py:72

Functiondict_flatten

Flatten a nested dictionary into a dictionary with no nested dictionaries.

trellis/utils/general_utils.py:102

Functiondict_foreach

Recursively apply a function to all non-dictionary leaf values in a dictionary.

trellis/utils/general_utils.py:35

Functiondict_merge

Merge two dictionaries.

trellis/utils/general_utils.py:28

Functiondict_reduce

Reduce a list of dictionaries. Leaf values must be scalars.

trellis/utils/general_utils.py:51

Functiondisable_sage_attention

Restore PyTorch's original scaled_dot_product_attention function.

trellis/modules/attention_utils.py:20

Methoddtype

(self)

trellis/modules/sparse/basic.py:176

Functionenable_sage_attention

Enable SageAttention by replacing PyTorch's scaled_dot_product_attention with sageattn from the SageAttention library.

trellis/modules/attention_utils.py:12

Methodextract_luminosity_difference

(self, original_image, processed_image, amplification=1.0)

StableXWrapper.py:153

Methodfeats

(self)

trellis/modules/sparse/basic.py:148

Methodforward

(self, x: torch.Tensor)

trellis/modules/norm.py:14

Methodforward

(self, x: torch.Tensor)

trellis/modules/norm.py:19

Methodforward

(self, x: torch.Tensor)

trellis/modules/attention/modules.py:14

Methodforward

Args: q (sp.SparseTensor): [..., N, D] tensor of queries k (sp.SparseTensor): [..., N, D] tensor of keys

trellis/modules/attention/modules.py:40

Methodforward

(self, x: torch.Tensor, context: Optional[torch.Tensor] = None, indices: Optional[torch.Tensor] = None)

trellis/modules/attention/modules.py:112

Methodforward

(self, input: SparseTensor)

trellis/modules/sparse/linear.py:14

Methodforward

(self, input: SparseTensor)

trellis/modules/sparse/spatial.py:22

Methodforward

(self, input: SparseTensor)

trellis/modules/sparse/spatial.py:68

Methodforward

(self, input: SparseTensor)

trellis/modules/sparse/spatial.py:92

Methodforward

(self, input: SparseTensor)

trellis/modules/sparse/norm.py:35

Methodforward

(self, x: SparseTensor)

trellis/modules/sparse/norm.py:50

Methodforward

(self, x: SparseTensor)

trellis/modules/sparse/norm.py:57

Methodforward

(self, input: SparseTensor)

trellis/modules/sparse/nonlinearity.py:19

Methodforward

(self, input: SparseTensor)

trellis/modules/sparse/nonlinearity.py:24

Methodforward

(self, input: SparseTensor)

trellis/modules/sparse/nonlinearity.py:33

Methodforward

(self, x: Union[SparseTensor, torch.Tensor])

trellis/modules/sparse/attention/modules.py:18

Methodforward

(self, x: Union[SparseTensor, torch.Tensor], context: Optional[Union[SparseTensor, torch.Tensor]] = None)

trellis/modules/sparse/attention/modules.py:105

Methodforward

(self, x: SparseTensor)

trellis/modules/sparse/conv/conv_torchsparse.py:18

Methodforward

(self, x: SparseTensor)

trellis/modules/sparse/conv/conv_torchsparse.py:34

Methodforward

(self, x: SparseTensor)

trellis/modules/sparse/conv/conv_spconv.py:29

Methodforward

(self, x: SparseTensor)

trellis/modules/sparse/conv/conv_spconv.py:65

Methodforward

(self, x: SparseTensor)

trellis/modules/sparse/transformer/blocks.py:20

Methodforward

(self, x: SparseTensor)

trellis/modules/sparse/transformer/blocks.py:74

Methodforward

(self, x: SparseTensor, context: torch.Tensor)

trellis/modules/sparse/transformer/blocks.py:147

Methodforward

(self, x: SparseTensor, mod: torch.Tensor)

trellis/modules/sparse/transformer/modulated.py:74

Methodforward

(self, x: SparseTensor, mod: torch.Tensor, context: torch.Tensor)

trellis/modules/sparse/transformer/modulated.py:162

Methodforward

Args: x (torch.Tensor): (N, D) tensor of spatial positions

trellis/modules/transformer/blocks.py:35

Methodforward

(self, x: torch.Tensor)

trellis/modules/transformer/blocks.py:58

Methodforward

(self, x: torch.Tensor)

trellis/modules/transformer/blocks.py:108

Methodforward

(self, x: torch.Tensor, context: torch.Tensor)

trellis/modules/transformer/blocks.py:177

Methodforward

(self, x: torch.Tensor, mod: torch.Tensor)

trellis/modules/transformer/modulated.py:69

Methodforward

(self, x: torch.Tensor, mod: torch.Tensor, context: torch.Tensor)

trellis/modules/transformer/modulated.py:152

Methodforward

(self, t)

trellis/models/sparse_structure_flow.py:49

Methodforward

(self, x: torch.Tensor, t: torch.Tensor, cond: torch.Tensor)

trellis/models/sparse_structure_flow.py:176

Methodforward

(self, x: torch.Tensor)

trellis/models/sparse_structure_vae.py:39

Methodforward

(self, x: torch.Tensor)

trellis/models/sparse_structure_vae.py:68

Methodforward

(self, x: torch.Tensor)

trellis/models/sparse_structure_vae.py:93

Methodforward

(self, x: torch.Tensor, sample_posterior: bool = False, return_raw: bool = False)

trellis/models/sparse_structure_vae.py:186

Methodforward

(self, x: torch.Tensor)

trellis/models/sparse_structure_vae.py:295

Methodforward

(self, x: sp.SparseTensor, emb: torch.Tensor)

trellis/models/structured_latent_flow.py:52

← previous next →201–300 of 333, ranked by callers