Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/AMAP-ML/Eevee
/ functions
Functions
226 in github.com/AMAP-ML/Eevee
⨍
Functions
226
◇
Types & classes
53
↓ 62 callers
Method
to
(self, *args, **kwargs)
models/pipeline.py:157
↓ 17 callers
Method
__init__
( self, in_channels: int, out_channels: int, factor_t, factor_s=1,
models/vae.py:399
↓ 17 callers
Function
hash_state_dict_keys
(state_dict, with_shape=True)
models/dit.py:44
↓ 16 callers
Method
encode
(self, x, scale)
models/vae.py:984
↓ 10 callers
Method
preprocess_video
(self, video, torch_dtype=None, device=None, pattern="B C T H W", min_value=-1, max_value=1)
models/pipeline.py:128
↓ 8 callers
Method
__init__
(self, in_dim, out_dim, has_pos_emb=False)
models/dit.py:257
↓ 7 callers
Function
check_is_instance
(model, module_class)
models/vae.py:11
↓ 6 callers
Method
clear_cache
(self)
models/vae.py:1048
↓ 6 callers
Method
load_models_to_device
(self, model_names=[])
models/pipeline.py:106
↓ 6 callers
Method
to
(self, *args, **kwargs)
models/training.py:198
↓ 5 callers
Method
__init__
(self, dim, eps=1e-6)
models/text_encoder.py:24
↓ 5 callers
Function
create_custom_forward
(module)
models/pipeline.py:373
↓ 5 callers
Method
norm
(self, x)
models/dit.py:129
↓ 4 callers
Method
decode
(self, z, scale)
models/vae.py:1011
↓ 4 callers
Method
fetch_model
(self, model_name)
models/manager.py:152
↓ 4 callers
Method
from_pretrained
( torch_dtype = torch.bfloat16, device = "cuda", vae_model_path = None, text_e
models/pipeline.py:39
↓ 3 callers
Method
__getitem__
(self, item)
test/test.py:61
↓ 3 callers
Function
basic_clean
(text)
models/prompter.py:47
↓ 3 callers
Method
load_model
(self, file_path, model_type=None, device=None, torch_dtype=None)
models/manager.py:123
↓ 3 callers
Function
load_state_dict
(file_path, torch_dtype=None, device="cpu")
models/manager.py:12
↓ 3 callers
Function
precompute_freqs_cis
(dim: int, end: int = 1024, theta: float = 10000.0)
models/dit.py:105
↓ 3 callers
Method
process_csv
(self, csv_path, dataset_base_path)
dataset/eevee_dataset.py:63
↓ 3 callers
Method
process_video
(self, file_path)
dataset/eevee_dataset.py:79
↓ 3 callers
Method
set_timesteps
(self, num_inference_steps=100, denoising_strength=1.0, training=False, shift=None, dynamic_shift_len=None, ex
models/scheduler.py:32
↓ 3 callers
Method
state_dict_converter
()
models/dit.py:432
↓ 3 callers
Method
step
(self, model_output, timestep, sample, to_final=False, **kwargs)
models/scheduler.py:70
↓ 2 callers
Method
__len__
(self)
test/test.py:48
↓ 2 callers
Method
build_1d_mask
(self, length, left_bound, right_bound, border_width)
models/vae.py:1081
↓ 2 callers
Method
build_mask
(self, data, is_bound, border_width)
models/vae.py:1090
↓ 2 callers
Method
check_resize_height_width
(self, height, width, num_frames=None)
models/pipeline.py:83
↓ 2 callers
Function
count_conv3d
(model)
models/vae.py:943
↓ 2 callers
Method
encode_prompt
(self, prompt, positive=True, device="cuda")
models/prompter.py:135
↓ 2 callers
Method
export_trainable_state_dict
(self, state_dict, remove_prefix=None)
models/training.py:203
↓ 2 callers
Method
fetch_tokenizer
(self, tokenizer_path=None)
models/prompter.py:128
↓ 2 callers
Function
flash_attention
(q: torch.Tensor, k: torch.Tensor, v: torch.Tensor, num_heads: int, compatibility_mode=False)
models/dit.py:50
↓ 2 callers
Method
forward
(self, x: torch.Tensor, first_chunk=False)
models/vae.py:417
↓ 2 callers
Function
fp16_clamp
(x)
models/text_encoder.py:8
↓ 2 callers
Method
generate_noise
(self, shape, seed=None, rand_device="cpu", rand_torch_dtype=torch.float32, device=None, torch_dtype=None)
models/pipeline.py:99
↓ 2 callers
Method
image_crop_and_resize
(self, image)
dataset/eevee_dataset.py:91
↓ 2 callers
Method
load
(self, model: torch.nn.Module, state_dict_lora, alpha=1.0)
models/pipeline.py:418
↓ 2 callers
Function
modulate
(x: torch.Tensor, shift: torch.Tensor, scale: torch.Tensor)
models/dit.py:86
↓ 2 callers
Method
patchify
(self, x: torch.Tensor, control_camera_latents_input: Optional[torch.Tensor] = None)
models/dit.py:359
↓ 2 callers
Method
prompt_embedder
(self, inputs)
models/pipeline.py:283
↓ 2 callers
Function
rope_apply
(x, freqs, num_heads)
models/dit.py:114
↓ 2 callers
Method
save_model
(self, accelerator, model, file_name)
utils/logger.py:33
↓ 2 callers
Function
sinusoidal_embedding_1d
(dim, position)
models/dit.py:90
↓ 2 callers
Method
unpatchify
(self, x: torch.Tensor, grid_size: torch.Tensor)
models/dit.py:369
↓ 2 callers
Function
whitespace_clean
(text)
models/prompter.py:53
↓ 1 callers
Method
__init__
( self, vace_layers=(0, 2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 22, 24, 26, 28), vace_in_d
models/vace.py:50
↓ 1 callers
Method
_clean
(self, text)
models/prompter.py:110
↓ 1 callers
Method
_relative_position_bucket
(self, rel_pos)
models/text_encoder.py:171
↓ 1 callers
Method
add_lora_to_model
( self, model, target_modules, lora_rank, upcast_d
models/training.py:80
↓ 1 callers
Method
add_noise
(self, original_samples, noise, timestep)
models/scheduler.py:92
↓ 1 callers
Method
calculate_shift
( self, image_seq_len, base_seq_len: int = 256, max_seq_len: int = 8192,
models/scheduler.py:112
↓ 1 callers
Function
canonicalize
(text, keep_punctuation_exact_string=None)
models/prompter.py:59
↓ 1 callers
Function
convert_state_dict_keys_to_single_str
(state_dict, with_shape=True)
models/dit.py:29
↓ 1 callers
Function
convert_state_dict_keys_to_single_str
(state_dict, with_shape=True)
models/vace.py:5
↓ 1 callers
Method
fetch_models
(self, text_encoder: WanTextEncoder = None)
models/prompter.py:132
↓ 1 callers
Method
forward
( self, x, vace_context, context, t_mod, freqs, use_gradient_checkpointing: bool = False,
models/vace.py:75
↓ 1 callers
Method
forward_preprocess
(self, data)
models/training.py:95
↓ 1 callers
Method
freeze_except
(self, model_names)
models/pipeline.py:74
↓ 1 callers
Method
from_civitai
(self, state_dict)
models/dit.py:517
↓ 1 callers
Method
get_name_dict
(self, lora_state_dict)
models/pipeline.py:401
↓ 1 callers
Function
hash_state_dict_keys
(state_dict, with_shape=True)
models/vace.py:20
↓ 1 callers
Function
init_weights_on_device
(device = torch.device("meta"), include_buffers :bool = False)
models/manager.py:39
↓ 1 callers
Method
input_video_embedder
(self, inputs)
models/pipeline.py:289
↓ 1 callers
Method
input_video_embedder
(self, inputs)
models/training.py:149
↓ 1 callers
Function
launch_training_task
( dataset, model, model_logger, learning_rate = 1e-5, weight_decay = 1e-2, num_workers
utils/launch.py:7
↓ 1 callers
Method
load_lora
( self, module, lora_config = None, alpha = 1 )
models/pipeline.py:166
↓ 1 callers
Function
load_model_from_single_file
(state_dict, model_names, model_classes, torch_dtype, device)
models/manager.py:87
↓ 1 callers
Function
load_state_dict_from_bin
(file_path, torch_dtype=None, device="cpu")
models/manager.py:29
↓ 1 callers
Function
load_state_dict_from_safetensors
(file_path, torch_dtype=None, device="cpu")
models/manager.py:19
↓ 1 callers
Method
noise_initializer
(self, inputs)
models/pipeline.py:275
↓ 1 callers
Method
noise_initializer
(self, inputs)
models/training.py:133
↓ 1 callers
Method
on_epoch_end
(self, accelerator, model, epoch_id)
utils/logger.py:17
↓ 1 callers
Method
on_step_end
(self, accelerator, model, save_steps=None)
utils/logger.py:11
↓ 1 callers
Method
on_training_end
(self, accelerator, model, save_steps=None)
utils/logger.py:28
↓ 1 callers
Function
parse_args
()
utils/parser.py:4
↓ 1 callers
Function
patch_tensor_constructor
(fn)
models/manager.py:57
↓ 1 callers
Function
patchify
(x, patch_size)
models/vae.py:199
↓ 1 callers
Function
precompute_freqs_cis_3d
(dim: int, end: int = 1024, theta: float = 10000.0)
models/dit.py:97
↓ 1 callers
Method
preprocess_image
(self, image, torch_dtype=None, device=None, pattern="B C H W", min_value=-1, max_value=1)
models/pipeline.py:134
↓ 1 callers
Method
process_image
(self, file_path)
dataset/eevee_dataset.py:73
↓ 1 callers
Method
process_prompt
(self, prompt, positive=True)
models/prompter.py:31
↓ 1 callers
Method
prompt_embedder
(self, inputs)
models/training.py:142
↓ 1 callers
Method
reparameterize
(self, mu, log_var)
models/vae.py:1036
↓ 1 callers
Function
save_video
(frames, save_path, fps, quality=9, ffmpeg_params=None)
test/test.py:78
↓ 1 callers
Method
set_shape
(self, height, width)
test/test.py:44
↓ 1 callers
Method
shape_checker
(self, inputs)
models/pipeline.py:268
↓ 1 callers
Method
shape_checker
(self, inputs)
models/training.py:125
↓ 1 callers
Method
single_decode
(self, hidden_state, device)
models/vae.py:1212
↓ 1 callers
Method
single_encode
(self, video, device)
models/vae.py:1206
↓ 1 callers
Method
switch_pipe_to_training_mode
( self, pipe, lora_base_model, # "vace" lora_target_modules, # "q,k,
models/training.py:57
↓ 1 callers
Method
tiled_decode
(self, hidden_states, device, tile_size, tile_stride)
models/vae.py:1103
↓ 1 callers
Method
tiled_encode
(self, video, device, tile_size, tile_stride)
models/vae.py:1155
↓ 1 callers
Method
trainable_modules
(self)
models/training.py:193
↓ 1 callers
Method
trainable_param_names
(self)
models/training.py:215
↓ 1 callers
Method
training_loss
(self, **inputs)
models/pipeline.py:142
↓ 1 callers
Method
training_target
(self, sample, noise, timestep)
models/scheduler.py:101
↓ 1 callers
Method
training_weight
(self, timestep)
models/scheduler.py:106
next →
1–100 of 226, ranked by callers