↓ 1 callersMethodalign_trajectory(self, pipe: BasePipeline, timesteps_teacher, trajectory_teacher, inputs_shared, inputs_posi, inputs_nega, num
game_editing/diffsynth/diffusion/loss.py:70
↓ 1 callersFunctionapply_multimodal_rotary_pos_emb(q, k, cos, sin, mrope_section, unsqueeze_dim=1)
game_editing/diffsynth/models/nexus_gen_projector.py:15
↓ 1 callersMethodauto_load_model(self, path, vram_config=None, vram_limit=None, clear_parameters=False, state_dict=None)
game_editing/diffsynth/models/model_loader.py:63
↓ 1 callersMethodblockwise_forward(self, image, conditionings: list[torch.Tensor], controlnet_inputs: list[ControlNetInput], progress_id, num_in
game_editing/diffsynth/pipelines/qwen_image.py:216
↓ 1 callersMethodcompute_freqs_mot(self, f, h, w, end: int = 1024, theta: float = 10000.0)
game_editing/diffsynth/models/wan_video_mot.py:146
↓ 1 callersMethodcompute_regularization(self, pipe: BasePipeline, trajectory_teacher, inputs_shared, inputs_posi, inputs_nega, num_inference_steps, c
game_editing/diffsynth/diffusion/loss.py:98
↓ 1 callersFunctionenable_vram_management_recursively(model: torch.nn.Module, module_map: dict, vram_config: dict, vram_limit=None, name_prefix="", disk_map=None,
game_editing/diffsynth/core/vram/layers.py:439
↓ 1 callersMethodencode_prompt(
self,
text_encoder,
tokenizer,
prompt: Union[str, List[str]],
dtype
game_editing/diffsynth/pipelines/flux2_image.py:253
↓ 1 callersMethodencode_prompt(
self,
tokenizer_1,
tokenizer_2,
text_encoder_1,
text_encoder_2,
game_editing/diffsynth/pipelines/flux_image.py:373
↓ 1 callersMethodencode_prompt(
self,
tokenizer_1,
tokenizer_2,
text_encoder_1,
text_encoder_2,
game_editing/diffsynth/pipelines/flux_image.py:554
↓ 1 callersMethodencode_prompt_omni(
self,
pipe,
prompt: Union[str, List[str]],
edit_image=None,
device:
game_editing/diffsynth/pipelines/z_image.py:233
↓ 1 callersMethodencode_prompt_using_clip(self, prompt, text_encoder, tokenizer, max_length, device)
game_editing/diffsynth/pipelines/flux_image.py:351
↓ 1 callersMethodencode_prompt_using_clip(self, prompt, text_encoder, tokenizer, max_length, device)
game_editing/diffsynth/pipelines/flux_image.py:532
↓ 1 callersMethodencode_prompt_using_t5(self, prompt, text_encoder, tokenizer, max_length, device)
game_editing/diffsynth/pipelines/flux_image.py:362
↓ 1 callersMethodencode_prompt_using_t5(self, prompt, text_encoder, tokenizer, max_length, device)
game_editing/diffsynth/pipelines/flux_image.py:543
↓ 1 callersMethodextract_audio_feat(self, input_audio, sample_rate, processor, return_all_layers=False, dtype=torch.float32, device='cpu')
game_editing/diffsynth/models/wav2vec.py:102
↓ 1 callersMethodfetch_trajectory(self, pipe: BasePipeline, timesteps_student, inputs_shared, inputs_posi, inputs_nega, num_inference_steps, cf
game_editing/diffsynth/diffusion/loss.py:53
↓ 1 callersFunctionflash_attention_2(q: torch.Tensor, k: torch.Tensor, v: torch.Tensor, q_pattern="b n s d", k_pattern="b n s d", v_pattern="b n s
game_editing/diffsynth/core/attention/attention.py:84