Muhammad Taqi Raza commited on
Commit
33667dd
·
1 Parent(s): 34bf23a

adding some prints

Browse files
inference/cli_demo_camera_i2v_pcd.py CHANGED
@@ -427,7 +427,7 @@ if __name__ == "__main__":
427
  parser.add_argument("--num_frames", type=int, default=49)
428
  parser.add_argument("--start_camera_idx", type=int, default=0)
429
  parser.add_argument("--end_camera_idx", type=int, default=1)
430
- parser.add_argument("--controlnet_transformer_num_attn_heads", type=int, default=2)
431
  parser.add_argument("--controlnet_transformer_attention_head_dim", type=int, default=64)
432
  parser.add_argument("--controlnet_transformer_out_proj_dim_factor", type=int, default=64)
433
  parser.add_argument("--controlnet_transformer_out_proj_dim_zero_init", action="store_true", default=True, help=("Init project zero."),
 
427
  parser.add_argument("--num_frames", type=int, default=49)
428
  parser.add_argument("--start_camera_idx", type=int, default=0)
429
  parser.add_argument("--end_camera_idx", type=int, default=1)
430
+ parser.add_argument("--controlnet_transformer_num_attn_heads", type=int, default=4)
431
  parser.add_argument("--controlnet_transformer_attention_head_dim", type=int, default=64)
432
  parser.add_argument("--controlnet_transformer_out_proj_dim_factor", type=int, default=64)
433
  parser.add_argument("--controlnet_transformer_out_proj_dim_zero_init", action="store_true", default=True, help=("Init project zero."),