DETECT_SAM/efficient_sam.py 报错

DETECT_SAM/efficient_sam.py报错，TorchScript相关的错误

报错日志：

Traceback (most recent call last):
  File "AutoStudio/run.py", line 231, in <module>
    output = autostudio.generate(
  File "AutoStudio/model/autostudio.py", line 891, in generate
    seg_img, detection = process_image(detect_model=dino_model, same_model=same_model, input_image=i[1][0], categories=character_prompt_full, device='cuda:1') #if CUDA out of spa
  File "AutoStudio/DETECT_SAM/detectSam.py", line 108, in process_image
    detections.mask = inference_with_boxes(
  File "AutoStudio/DETECT_SAM/efficient_sam.py", line 62, in inference_with_boxes
    mask = inference_with_box(image, box, model, device)
  File "AutoStudio/DETECT_SAM/efficient_sam.py", line 31, in inference_with_box
    predicted_logits, predicted_iou = model(

  File "/data/condaenvs/autostudio/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)

RuntimeError: The following operation failed in the TorchScript interpreter.
Traceback of TorchScript (most recent call last):
RuntimeError: The following operation failed in the TorchScript interpreter.
Traceback of TorchScript (most recent call last):
RuntimeError: The following operation failed in the TorchScript interpreter.
Traceback of TorchScript (most recent call last):
RuntimeError: shape '[0, 3, 1, 2]' is invalid for input of size 1572864

donahowe / AutoStudio

DETECT_SAM/efficient_sam.py 报错 #23