Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Francis-Rings/StableAnimator
/ functions
Functions
480 in github.com/Francis-Rings/StableAnimator
⨍
Functions
480
◇
Types & classes
92
Method
forward
Args: x (torch.Tensor): image features shape (b, n1, D) latent (torch.Tensor): latent features
animation/modules/id_encoder.py:28
Method
forward
(self, latents, x)
animation/modules/id_encoder.py:100
Method
forward
(self, id_embeds, clip_embeds, shortcut=False, scale=1.0)
animation/modules/id_encoder.py:134
Method
forward
(self, imgs)
animation/helper/onnx_helper.py:206
Method
forward
Parameters: ---------- local_embeddings: torch.Tensor feature embeddings on each GPU(Rank). local_labels:
animation/helper/partial_fc_v2.py:106
Method
forward
animation/helper/partial_fc_v2.py:177
Method
forward
(self, logit_part, label_part)
animation/helper/partial_fc_v2.py:226
Method
forward
(ctx, tensor, *gather_list)
animation/helper/partial_fc_v2.py:234
Method
forward
(self, logits, labels)
animation/helper/losses.py:27
Method
forward
(self, logits: torch.Tensor, labels: torch.Tensor)
animation/helper/losses.py:74
Method
forward
(self, logits: torch.Tensor, labels: torch.Tensor)
animation/helper/losses.py:94
Method
forward
(self, x)
animation/helper/lr_scheduler.py:52
Method
forward
(self, x)
animation/helper/backbones/vit.py:31
Method
forward
(self, x)
animation/helper/backbones/vit.py:54
Method
forward
(self, x)
animation/helper/backbones/vit.py:107
Method
forward
(self, x)
animation/helper/backbones/vit.py:127
Method
forward
(self, x)
animation/helper/backbones/vit.py:277
Method
forward
(self, x)
animation/helper/backbones/iresnet.py:60
Method
forward
(self, x)
animation/helper/backbones/iresnet.py:148
Method
forward
(self, x)
animation/helper/backbones/iresnet2060.py:50
Method
forward
(self, x)
animation/helper/backbones/iresnet2060.py:151
Method
forward
(self, x)
animation/helper/backbones/mobilefacenet.py:12
Method
forward
(self, x)
animation/helper/backbones/mobilefacenet.py:25
Method
forward
(self, x)
animation/helper/backbones/mobilefacenet.py:37
Method
forward
(self, x)
animation/helper/backbones/mobilefacenet.py:51
Method
forward
(self, x)
animation/helper/backbones/mobilefacenet.py:71
Method
forward
(self, x)
animation/helper/backbones/mobilefacenet.py:84
Method
forward
(self, x)
animation/helper/backbones/mobilefacenet.py:134
Function
generate
( image_input: str, pose_input: str, width: int, height: int, guidance_scale: flo
app.py:87
Method
get_classifier
(self)
animation/helper/backbones/vit.py:221
Method
get_face_masks
(self, pil_img)
animation/dataset/animation_dataset.py:58
Function
get_image_pose
(ref_image_path)
DWPose/skeleton_extraction.py:181
Method
get_lr
(self)
animation/helper/lr_scheduler.py:14
Method
get_prepare_faceid
(self, face_image)
animation/pipelines/inference_pipeline_animation.py:113
Method
get_prepare_faceid
(self, face_image)
animation/pipelines/validation_pipeline_animation.py:141
Method
get_prepare_faceid
(self, face_image)
animation/pipelines/inference_pipeline_animation_pro.py:141
Method
get_velocity
(self, sample: torch.Tensor, noise: torch.Tensor, timesteps: torch.Tensor)
animation/pipelines/euler_discrete_pro.py:775
Method
guidance_scale
(self)
animation/pipelines/inference_pipeline_animation.py:329
Method
guidance_scale
(self)
animation/pipelines/pipeline_animation.py:350
Method
guidance_scale
(self)
animation/pipelines/validation_pipeline_animation.py:357
Method
guidance_scale
(self)
animation/pipelines/inference_pipeline_animation_pro.py:332
Function
handDetect
(candidate, subset, oriImg)
DWPose/dwpose_utils/util.py:155
Method
init_noise_sigma
(self)
animation/pipelines/euler_discrete_pro.py:338
Function
load_bin
(path, image_size)
animation/helper/eval/verification.py:200
Method
meta_info
(self)
animation/helper/onnx_helper.py:202
Method
no_weight_decay
(self)
animation/helper/backbones/vit.py:218
Function
npmax
(array)
DWPose/dwpose_utils/util.py:292
Method
num_timesteps
(self)
animation/pipelines/inference_pipeline_animation.py:342
Method
num_timesteps
(self)
animation/pipelines/pipeline_animation.py:363
Method
num_timesteps
(self)
animation/pipelines/validation_pipeline_animation.py:370
Method
num_timesteps
(self)
animation/pipelines/inference_pipeline_animation_pro.py:345
Function
padRightDownCorner
(img, stride, padValue)
DWPose/dwpose_utils/util.py:38
Function
patch_geglu_inplace
Patch GEGLU with inplace multiplication to save GPU memory.
animation/utils/geglu_patch.py:4
Function
rand_log_normal
Draws samples from an lognormal distribution.
train.py:212
Function
rand_log_normal
Draws samples from an lognormal distribution.
train_single.py:215
Function
read_image_feature
(path)
animation/helper/eval_ijbc.py:145
Function
read_image_feature
(path)
animation/helper/onnx_ijbc.py:94
Function
read_score
(path)
animation/helper/eval_ijbc.py:306
Function
read_worker
(args, q_in)
animation/helper/scripts/shuffle_rec.py:10
Method
release_memory
(self)
DWPose/dwpose_utils/dwpose_detector.py:15
Function
save_combined_frames
(batch_output, validation_images, validation_control_images,output_folder)
train.py:96
Function
save_combined_frames
(batch_output, validation_images, validation_control_images, output_folder)
train_single.py:99
Function
save_frames_as_mp4
(frames, output_mp4_path, fps)
inference_basic.py:47
Function
save_to_mp4
(frames, save_path, fps=7)
animation/utils/utils.py:81
Method
set_begin_index
Sets the begin index for the scheduler. This function should be run from pipeline before the inference. Args: begin_
animation/pipelines/euler_discrete_pro.py:361
Method
set_default_attn_processor
Disables custom attention processors and sets the default attention implementation.
animation/modules/unet.py:318
Method
set_default_attn_processor
Disables custom attention processors and sets the default attention implementation.
animation/modules/refined_vae.py:292
Method
set_ip_adapter
(self)
animation/pipelines/inference_pipeline_animation.py:122
Method
set_ip_adapter
(self)
animation/pipelines/validation_pipeline_animation.py:150
Method
set_timesteps
Sets the discrete timesteps used for the diffusion chain (to be run before inference). Args: num_inference_steps (`i
animation/pipelines/euler_discrete_pro.py:395
Function
smart_resize
(x, s)
DWPose/dwpose_utils/util.py:10
Function
smart_resize_k
(x, fx, fy)
DWPose/dwpose_utils/util.py:24
Method
step_index
The index counter for current timestep. It will increase 1 after each scheduler step.
animation/pipelines/euler_discrete_pro.py:347
Function
tensor2vid_scheduler
(video, processor, output_type="np")
animation/pipelines/euler_discrete_pro.py:76
Function
test
(data_set, backbone, batch_size, nfolds=10)
animation/helper/eval/verification.py:227
Function
transfer
(model, model_weights)
DWPose/dwpose_utils/util.py:61
Function
verification2
(template_norm_feats=None, unique_templates=None, p1=None,
animation/helper/eval_ijbc.py:282
Function
verification2
(template_norm_feats=None, unique_templates=None, p1=None,
animation/helper/onnx_ijbc.py:148
Function
worker_init_fn
(worker_id, num_workers, rank, seed)
animation/helper/utils/utils_distributed_sampler.py:25
Function
write_worker
(args, q_out)
animation/helper/scripts/shuffle_rec.py:30
← previous
401–480 of 480, ranked by callers