MCPcopy Create free account

hub / github.com/TMElyralab/MuseV / functions

Functions388 in github.com/TMElyralab/MuseV

↓ 1 callersFunctiondrop_last_repeat_context
if len(contexts)>=2 and the max value the oenultimate list same as of the last list Args: List (_type_): _description_ Returns:
musev/pipelines/context.py:105
↓ 1 callersFunctionencode_weighted_prompt
r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `list(int)`): prompt to be encoded
musev/utils/text_emb_util.py:352
↓ 1 callersFunctionformat_dct_texts
使用含有"{key}" 的模板对多属性文本字典进行拼接,形成新文本 concatenate multiple attribute text dictionaries using a template containing "{key}" to form a new text
musev/auto_prompt/attributes/attributes.py:144
↓ 1 callersFunctionformat_tuple_texts
使用含有"{}" 的模板对多属性文本元组进行拼接,形成新文本 concatenate multiple attribute text tuples using a template containing "{}" to form a new text Args:
musev/auto_prompt/attributes/attributes.py:129
↓ 1 callersMethodforward
(self, conditioning)
musev/models/controlnet.py:361
↓ 1 callersMethodfrom_pretrained_2d
r""" Instantiate a pretrained pytorch model from a pre-trained model configuration. The model is set in evaluation mode by default us
musev/models/unet_3d_condition.py:1284
↓ 1 callersFunctiongenerate_cuid
()
scripts/gradio/gradio_video2video.py:642
↓ 1 callersFunctiongenerate_cuid
()
scripts/gradio/gradio_text2video.py:610
↓ 1 callersFunctiongenerate_fix_two_stages_parameters
(start, stop, num, n_fix_start: int)
musev/utils/timesteps_util.py:48
↓ 1 callersFunctiongenerate_linear_parameters
(start, stop, num)
musev/utils/timesteps_util.py:30
↓ 1 callersFunctiongenerate_mask_from_indices
( shape: Tuple[int, int], indices: Tuple[np.ndarray, np.ndarray], big_value: float = 0, small_
musev/utils/attention_util.py:18
↓ 1 callersFunctiongenerate_parameters_with_timesteps
( start: int, num: int, stop: int = None, method: Literal["linear", "two_stage", "three_stage"
musev/utils/timesteps_util.py:5
↓ 1 callersFunctiongenerate_three_stages_parameters
(start, stop, num)
musev/utils/timesteps_util.py:55
↓ 1 callersFunctiongenerate_two_stages_parameters
(start, stop, num)
musev/utils/timesteps_util.py:41
↓ 1 callersFunctionget_2d_sincos_pos_embed
grid_size: int of the grid height and width return: pos_embed: [grid_size*grid_size, embed_dim] or [1+grid_size*grid_size, embed_dim] (w/ or
musev/models/embeddings.py:24
↓ 1 callersFunctionget_context_scheduler
(name: str)
musev/pipelines/context.py:69
↓ 1 callersMethodget_controlnet_emb
( self, run_controlnet, guess_mode, do_classifier_free_guidance, laten
musev/pipelines/pipeline_controlnet.py:1202
↓ 1 callersFunctionget_diags_indices
( shape: Union[int, Tuple[int, int]], k_min: int = 0, k_max: int = 0 )
musev/utils/attention_util.py:8
↓ 1 callersMethodget_facein_image_emb
( self, refer_face_image, device, dtype, batch_size, do_classifier_free_guidance )
musev/pipelines/pipeline_controlnet.py:516
↓ 1 callersMethodget_ip_adapter_face_emb
( self, refer_face_image, device, dtype, batch_size, do_classifier_free_guidance )
musev/pipelines/pipeline_controlnet.py:601
↓ 1 callersMethodget_ip_adapter_image_emb
( self, ip_adapter_image, device, dtype, batch_size, do_classi
musev/pipelines/pipeline_controlnet.py:686
↓ 1 callersFunctionget_noise_pred_single
(latents, t, context, unet)
musev/utils/util.py:319
↓ 1 callersMethodget_referencenet_emb
( self, refer_image_vae_emb, refer_image, batch_size, num_videos_per_p
musev/pipelines/pipeline_controlnet.py:867
↓ 1 callersMethodget_referencenet_image_vae_emb
( self, refer_image, batch_size, num_videos_per_prompt, device,
musev/pipelines/pipeline_controlnet.py:782
↓ 1 callersMethodget_self_attns
r""" Returns: `dict` of attention attns: A dictionary containing all attention attns used in the model with indexed by
musev/models/referencenet.py:1168
↓ 1 callersFunctionget_template_by_name
根据 template_name 确定 prompt 生成器类 choose prompt generator class according to template_name Args: name (str): template 的名字简称,便于指定. te
musev/auto_prompt/load_template.py:10
↓ 1 callersFunctionget_weighted_text_embeddings
r""" Prompts can be assigned with local weights using brackets. For example, prompt 'A (very beautiful) masterpiece' highlights the words 'ver
musev/utils/text_emb_util.py:218
↓ 1 callersFunctionhack_t2i_sd_layer_attn_with_ip
( unet: nn.Module, self_attn_class: BaseIPAttnProcessor = None, cross_attn_class: BaseIPAttnProces
musev/models/unet_3d_condition.py:116
↓ 1 callersMethodhist_match_with_vis_cond
video: b c t1 h w target: b c t2(=1) h w
musev/pipelines/pipeline_controlnet.py:506
↓ 1 callersFunctioninit_prompt
(prompt, pipeline)
musev/utils/util.py:264
↓ 1 callersMethodinsert_spatial_self_attn_idx
(self)
musev/models/unet_3d_condition.py:1663
↓ 1 callersMethodinsert_spatial_self_attn_idx
(self)
musev/models/referencenet.py:1145
↓ 1 callersFunctionload_referencenet
Loads the ReferenceNet model. Args: sd_referencenet_model (Tuple[str, nn.Module] or str): The pretrained ReferenceNet model or the p
musev/models/referencenet_loader.py:43
↓ 1 callersFunctionload_yaml
(path)
scripts/inference/video2video.py:923
↓ 1 callersFunctionload_yaml
(path)
scripts/inference/text2video.py:808
↓ 1 callersFunctionmake_grid_with_opencv
read tensor batch and make a grid with opencv Args: batch (Union[torch.Tensor, np.ndarray]): 4 dim tensor, like b c h w nrows (in
musev/utils/util.py:78
↓ 1 callersFunctionmerge_multi_attrtext
对多属性文本元组进行拼接,形成新文本。 如果`template`含有{key},则根据key来取值; 如果`template`有且只有1个{},则根据先后顺序对texts中的值进行拼接。 concatenate multiple attribute
musev/auto_prompt/attributes/attributes.py:158
↓ 1 callersMethodmultistep_dpm_solver_second_order_update
One step for the second-order multistep DPM-Solver. Args: model_output_list (`List[torch.FloatTensor]`):
musev/schedulers/scheduling_dpmsolver_multistep.py:499
↓ 1 callersMethodmultistep_dpm_solver_third_order_update
One step for the third-order multistep DPM-Solver. Args: model_output_list (`List[torch.FloatTensor]`):
musev/schedulers/scheduling_dpmsolver_multistep.py:595
↓ 1 callersFunctionnext_step
( model_output: Union[torch.FloatTensor, np.ndarray], timestep: int, sample: Union[torch.FloatTens
musev/utils/util.py:287
↓ 1 callersFunctionparse_args
()
scripts/inference/video2video.py:54
↓ 1 callersFunctionparse_args
()
scripts/inference/text2video.py:54
↓ 1 callersFunctionparse_prompt_attention
Parses a string with attention tokens and returns a list of pairs: text and its associated weight. Accepted tokens are: (abc) - increas
musev/utils/text_emb_util.py:32
↓ 1 callersMethodprepare_condition_latents_and_index
( self, condition_images, condition_latents, video_length, batch_size,
musev/pipelines/pipeline_controlnet.py:966
↓ 1 callersMethodprepare_controlnet_and_guidance_parameter
( self, control_guidance_start, control_guidance_end )
musev/pipelines/pipeline_controlnet.py:1050
↓ 1 callersMethodprepare_controlnet_guess_mode
(self, controlnet, guess_mode)
musev/pipelines/pipeline_controlnet.py:1083
↓ 1 callersMethodprepare_controlnet_image_and_latents
( self, controlnet, width, height, batch_size, num_videos_per_
musev/pipelines/pipeline_controlnet.py:1092
↓ 1 callersFunctionprepare_global_context
( context_schedule: str, num_inference_steps: int, time_size: int, context_frames: int, co
musev/pipelines/context.py:120
↓ 1 callersMethodprepare_latents
支持多种情况下的latens: img_based_latents: 当Image t=1,latents=None时,使用image赋值到shape,然后加噪;适用于text2video、middle2video。 video_based_late
musev/pipelines/pipeline_controlnet.py:240
↓ 1 callersMethodpreset
(self)
musev/auto_prompt/attributes/attributes.py:225
↓ 1 callersFunctionrandom_noise
( tensor: torch.Tensor = None, shape: Tuple[int] = None, dtype: torch.dtype = None, device: to
musev/utils/noise_util.py:8
↓ 1 callersFunctionread_image
(path)
scripts/inference/video2video.py:957
↓ 1 callersFunctionread_image
(path)
scripts/inference/text2video.py:842
↓ 1 callersFunctionread_image
(path)
scripts/gradio/gradio_video2video.py:472
↓ 1 callersFunctionread_image
(path)
scripts/gradio/gradio_text2video.py:444
↓ 1 callersFunctionread_image_lst
(path)
scripts/inference/video2video.py:963
↓ 1 callersFunctionread_image_lst
(path)
scripts/inference/text2video.py:848
↓ 1 callersFunctionread_image_lst
(path)
scripts/gradio/gradio_video2video.py:478
↓ 1 callersFunctionread_image_lst
(path)
scripts/gradio/gradio_text2video.py:450
↓ 1 callersFunctionsample_by_idx
given a int to represent candidate list, sample n_sample with sample_rate from the candidate list Args: T (int): _description_ n_
musev/data/data_util.py:58
↓ 1 callersFunctionsave_videos_to_images
save video batch to images into image_type Args: videos (np.array): [h w c] path (str): image directory path
musev/utils/util.py:19
↓ 1 callersMethodset_attention_slice
r""" Enable sliced attention computation. When this option is enabled, the attention module will split the input tensor in slices, to
musev/models/unet_3d_condition.py:642
↓ 1 callersMethodset_timesteps
Sets the timesteps used for the diffusion chain. Supporting function to be run before inference. Args: num_inference_ste
musev/schedulers/scheduling_dpmsolver_multistep.py:237
↓ 1 callersFunctionsplit_index
_summary_ Args: indexs (List): _description_ n_first (int): _description_ n_last (int): _description_ method (Lit
musev/data/data_util.py:323
↓ 1 callersMethodstep
Predict the sample from the previous timestep by reversing the SDE. This function propagates the diffusion process from the learned m
musev/schedulers/scheduling_lcm.py:83
↓ 1 callersFunctionupdate_controlnet_processor_params
merge dst into src
musev/pipelines/pipeline_controlnet_predictor.py:82
↓ 1 callersFunctionupdate_pipeline_lora_model
使用 lora 更新pipeline中的unet相关参数 Args: pipeline (DiffusionPipeline): _description_ lora (Union[str, Dict]): _description_ alp
musev/utils/model_util.py:108
↓ 1 callersFunctionupdate_pipeline_model_parameters
( pipeline: DiffusionPipeline, model_path: str = None, lora_dict: Dict[str, Dict] = None, text
musev/utils/model_util.py:23
↓ 1 callersMethodupdate_string
(self, new_choice)
scripts/gradio/app_docker_space.py:140
↓ 1 callersMethodupdate_string
(self, new_choice)
scripts/gradio/app_gradio_space.py:173
↓ 1 callersMethodupdate_string
(self, new_choice)
scripts/gradio/app.py:138
↓ 1 callersFunctionupdate_unet_ip_adapter_cross_attn_param
use independent ip_adapter attn 中的 to_k, to_v in unet ip_adapter: like ['1.to_k_ip.weight', '1.to_v_ip.weight', '3.to_k_ip.weight'] Args:
musev/models/ip_adapter_face_loader.py:147
↓ 1 callersFunctionzero_module
(module)
musev/models/controlnet.py:319
Method__call__
r""" 旨在兼容text2video、text2image、img2img、video2video、是否有controlnet等的通用pipeline。目前仅不支持img2img、video2video。 支持多片段同时denoise,交叉部分加权平均
musev/pipelines/pipeline_controlnet.py:1295
Method__call__
(self, attributes: str)
musev/auto_prompt/attributes/render.py:25
Method__call__
(self, attributes: str)
musev/auto_prompt/attributes/human.py:109
Method__call__
将insight faces 检测的结果转化成prompt convert the results of insight faces detection to prompt Args: face_list (_type_): _desc
musev/auto_prompt/attributes/human.py:256
Method__call__
(self, attributes: str = None)
musev/auto_prompt/attributes/human.py:343
Method__call__
(self, attributes: str = None)
musev/auto_prompt/attributes/human.py:364
Method__call__
(self, attributes: str = None)
musev/auto_prompt/attributes/human.py:379
Method__call__
(self, attributes: str = None)
musev/auto_prompt/attributes/human.py:391
Method__call__
(self, attributes: str = None)
musev/auto_prompt/attributes/human.py:408
Method__call__
(self, attributes: str = None)
musev/auto_prompt/attributes/human.py:420
Method__call__
(self, attributes)
musev/auto_prompt/attributes/attributes.py:25
Method__call__
(self, attributes)
musev/auto_prompt/attributes/attributes.py:37
Method__call__
(self, attributes: str)
musev/auto_prompt/attributes/attributes.py:58
Method__call__
有时候一个属性可能会返回多个文本,如 style cartoon会返回宫崎骏和皮克斯两种风格,采用外积增殖成多个字典。 sometimes an attribute may return multiple texts, such as style cartoon w
musev/auto_prompt/attributes/attributes.py:86
Method__call__
(self, attributes: dict)
musev/auto_prompt/attributes/attr2template.py:67
Method__call__
(self, attributes: dict)
musev/auto_prompt/attributes/attr2template.py:126
Method__call__
( self, attn: IPAttention, hidden_states: torch.FloatTensor, encoder_hidden_st
musev/models/attention_processor.py:176
Method__call__
( self, attn: IPAttention, hidden_states: torch.FloatTensor, encoder_hidden_st
musev/models/attention_processor.py:378
Method__call__
( self, batch_size: int, device: str, dtype: torch.dtype, timesteps: L
musev/models/controlnet.py:93
Method__contains__
(self, key)
musev/utils/register.py:39
Method__getitem__
(self, key)
musev/utils/register.py:36
Method__init__
(self, options)
scripts/gradio/app_docker_space.py:136
Method__init__
(self, options)
scripts/gradio/app_gradio_space.py:169
Method__init__
(self, options)
scripts/gradio/app.py:134
Method__init__
(self, registry_name)
musev/utils/register.py:7
Method__init__
( self, sd_model_path: str, unet: nn.Module, controlnet_name: Union[str, List[
musev/pipelines/pipeline_controlnet_predictor.py:110
Method__init__
( self, vae: AutoencoderKL, unet: UNet3DConditionModel, scheduler: KarrasDiffu
musev/pipelines/pipeline_controlnet.py:160
← previousnext →101–200 of 388, ranked by callers