File "/Inf-DiT/generate_t2i_sr.py", line 211, in <module>
    main(args)
  
  File "/Inf-DiT/generate_t2i_sr.py", line 140, in main
    h, w = lr_image.shape[-2:]

AttributeError: 'Image' object has no attribute 'shape'