File size: 1,078 Bytes
0ccd807 3048ab4 0ccd807 3048ab4 0ccd807 3048ab4 0ccd807 3048ab4 0ccd807 |
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 |
{
"_class_name": "UNet3DModel",
"_diffusers_version": "0.30.3",
"_name_or_path": "workspace/tooncrafter/unet",
"addition_attention": true,
"attention_resolutions": [
4,
2,
1
],
"channel_mult": [
1,
2,
4,
4
],
"context_dim": 1024,
"conv_resample": true,
"default_fs": 24,
"dims": 2,
"dropout": 0.1,
"fs_condition": true,
"image_cross_attention": true,
"image_cross_attention_scale_learnable": false,
"in_channels": 8,
"model_channels": 320,
"normalize_range": 1.0,
"num_cond_trajs": 10,
"num_head_channels": 64,
"num_heads": -1,
"num_res_blocks": 2,
"out_channels": 4,
"resblock_updown": false,
"temporal_attention": true,
"temporal_conv": true,
"temporal_length": 16,
"temporal_selfatt_only": true,
"tempspatial_aware": false,
"traj_cond_attn": false,
"trajs_transform": "normalize",
"transformer_depth": 1,
"use_causal_attention": false,
"use_checkpoint": true,
"use_linear": true,
"use_relative_position": false,
"use_scale_shift_norm": false,
"with_traj_cond": false
}
|