File size: 16,679 Bytes
ad36971 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 |
2024-12-11 17:53:28,616 - __main__ - INFO - num_videos: 50
output_root_dir: outputs_video_video_prompts
output_dir: ${output_root_dir}/${now:%Y-%m-%d_%H-%M-%S}/benchmark_video_generation
generate_kwargs:
negative_prompt: ''
num_inference_steps: 50
guidance_scale: 6
num_frames: 45
prompt_sources:
- static/prompts/20241130_open_chatgpt4o_videos.txt
get_pipeline:
_target_: accediff.utils.factory.pipeline_factory.get_cogvideox_pipeline_with_tome_3d
repo_id: THUDM/CogVideoX-2b
hook_kwargs:
merge_ratio: 0.5
st: 4
sx: 6
sy: 9
tome_modele_names:
- transformer_blocks.{0..27}.attn1
2024-12-11 17:53:39,695 - bitsandbytes.cextension - DEBUG - Loading bitsandbytes native library from: /scratch/yz10381/miniconda3/envs/diffusers/lib/python3.10/site-packages/bitsandbytes/libbitsandbytes_cuda124.so
2024-12-11 17:53:44,380 - matplotlib - DEBUG - matplotlib data path: /scratch/yz10381/miniconda3/envs/diffusers/lib/python3.10/site-packages/matplotlib/mpl-data
2024-12-11 17:53:44,383 - matplotlib - DEBUG - CONFIGDIR=/home/yz10381/.config/matplotlib
2024-12-11 17:53:44,384 - matplotlib - DEBUG - interactive is False
2024-12-11 17:53:44,384 - matplotlib - DEBUG - platform is linux
2024-12-11 17:53:44,587 - matplotlib - DEBUG - CACHEDIR=/scratch/yz10381/.cache/matplotlib
2024-12-11 17:53:44,601 - matplotlib.font_manager - DEBUG - Using fontManager instance from /scratch/yz10381/.cache/matplotlib/fontlist-v390.json
2024-12-11 17:53:45,254 - urllib3.connectionpool - DEBUG - Starting new HTTPS connection (1): huggingface.co:443
2024-12-11 17:53:45,293 - urllib3.connectionpool - DEBUG - https://huggingface.co:443 "GET /api/models/THUDM/CogVideoX-2b HTTP/11" 200 1943
2024-12-11 17:53:45,326 - urllib3.connectionpool - DEBUG - https://huggingface.co:443 "HEAD /THUDM/CogVideoX-2b/resolve/main/model_index.json HTTP/11" 200 0
2024-12-11 17:53:52,159 - accediff.utils.factory.pipeline_factory - INFO - Available config keys: odict_keys(['num_attention_heads', 'attention_head_dim', 'in_channels', 'out_channels', 'flip_sin_to_cos', 'freq_shift', 'time_embed_dim', 'text_embed_dim', 'num_layers', 'dropout', 'attention_bias', 'sample_width', 'sample_height', 'sample_frames', 'patch_size', 'temporal_compression_ratio', 'max_text_seq_length', 'activation_fn', 'timestep_activation_fn', 'norm_elementwise_affine', 'norm_eps', 'spatial_interpolation_scale', 'temporal_interpolation_scale', 'use_rotary_positional_embeddings', 'use_learned_positional_embeddings', '_use_default_values', '_class_name', '_diffusers_version', '_name_or_path'])
2024-12-11 17:53:52,160 - accediff.utils.factory.pipeline_factory - INFO - f: 12, h: 30, w: 45, st: 4, sx: 6, sy: 9
2024-12-11 17:53:52,162 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.0.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,162 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.0.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,162 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.1.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,162 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.1.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,162 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.2.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,163 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.2.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.3.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.3.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.4.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.4.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.5.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.5.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.6.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.6.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.7.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.7.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.8.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.8.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.9.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.9.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.10.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.10.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.11.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.11.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.12.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.12.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.13.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.13.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.14.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.14.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.15.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.15.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.16.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.16.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.17.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.17.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.18.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.18.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.19.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,164 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.19.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.20.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.20.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.21.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.21.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.22.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.22.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.23.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.23.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.24.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.24.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.25.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.25.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.26.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.26.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.27.attn1 with pre forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,165 - accediff.utils.factory.pipeline_factory - INFO - Wrapping transformer_blocks.27.attn1 with post forward hook <function _get_cogvideox_tome_token_merging_hooks_3d.<locals>.tome_post_forward_hook at 0x14b09dac0e50>
2024-12-11 17:53:52,167 - __main__ - INFO - Saving videos to outputs_video_video_prompts/2024-12-11_17-53-28/benchmark_video_generation/videos
2024-12-11 17:53:52,168 - __main__ - INFO - Running iter_generate_prompt2video with 3 warmup iterations
2024-12-11 17:54:23,924 - __main__ - INFO - Warmup step 1: 31.091148614883423 seconds
2024-12-11 17:54:54,398 - __main__ - INFO - Warmup step 2: 30.020636320114136 seconds
2024-12-11 17:55:24,250 - __main__ - INFO - Warmup step 3: 29.390587091445923 seconds
2024-12-11 18:18:34,807 - __main__ - INFO - Total time: 1371.2745535373688 seconds for 47 iterations, throughput: 0.03427468254169656 iterations/second
|