runtime error
Exit code: 1. Reason: ear_projection', 'encoder_hid_dim_type', 'mid_block_type', 'dropout', 'time_cond_proj_dim', 'attention_type', 'projection_class_embeddings_input_dim', '_out_channels', 'addition_time_embed_dim', 'conv_in_kernel', 'time_embedding_act_fn', 'time_embedding_type', 'resnet_time_scale_shift', 'dual_cross_attention', '_center_input_sample', 'addition_embed_type', 'time_embedding_dim', 'num_class_embeds', '_landmark_net', 'encoder_hid_dim', 'transformer_layers_per_block', 'reverse_transformer_layers_per_block'} was not found in config. Values will be initialized to default values. Some weights of the model checkpoint were not used when initializing UNet2DConditionModel: ['conv_norm_out.bias, conv_norm_out.weight, conv_out.bias, conv_out.weight'] 12/06/2024 11:36:42 - INFO - joyhallo.models.unet_3d - loaded temporal unet's pretrained weights from pretrained_models/stable-diffusion-v1-5/unet ... The config attributes {'center_input_sample': False} were passed to UNet3DConditionModel, but are not expected and will be ignored. Please verify your config.json configuration file. {'dual_cross_attention', 'class_embed_type', 'only_cross_attention', 'upcast_attention', 'num_class_embeds', 'use_linear_projection', 'resnet_time_scale_shift'} was not found in config. Values will be initialized to default values. Load motion module params from pretrained_models/motion_module/mm_sd_v15_v2.ckpt 12/06/2024 11:36:58 - INFO - joyhallo.models.unet_3d - Loaded 453.20928M-parameter motion module loaded weight from ./pretrained_models/joyhallo/net.pth Traceback (most recent call last): File "/home/user/app/app.py", line 14, in <module> from scripts.inference import predict File "/home/user/app/scripts/inference.py", line 654, in <module> accelerator, vae, net, val_noise_scheduler, cfg, validation_dir = get_model(args) File "/usr/local/lib/python3.10/site-packages/spaces/zero/wrappers.py", line 214, in gradio_handler raise res.value _pickle.PicklingError: Can't pickle local object 'get_constant_schedule.<locals>.<lambda>'
Container logs:
Fetching error logs...