↓ 10 callersMethod__init__(self, embed_dim=1536, num_layers=24, use_rms_norm=False, num_dual_blocks=0, pos_embed_max_size=192)
DiffSynth-Studio/diffsynth/models/sd3_dit.py:343
↓ 9 callersMethod__init__(
self,
base_dim: int = 96,
z_dim: int = 16,
dim_mult: Tuple[int] = [1, 2, 4,
DiffSynth-Studio/diffsynth/models/qwen_image_vae.py:641
↓ 9 callersFunctionweighted_mean(x: torch.Tensor, w: torch.Tensor = None, dim: Union[int, torch.Size] = None, keepdim: bool = False, eps: floa
models/spatracker/models/moge/utils/geometry_torch.py:16
↓ 8 callersMethod__init__(self, input_ids, cu_input_ids, attention_mask, cu_seqlens, max_seq_len)
DiffSynth-Studio/diffsynth/models/stepvideo_text_encoder.py:185