NameError: name ‘_C‘ is not defined

AI算法网奇

于 2024-11-11 12:38:22 发布

阅读量480

点赞数 1

CC 4.0 BY-SA版权

分类专栏： python基础文章标签： python

本文链接：https://blog.csdn.net/jacke121/article/details/143679322

python基础专栏收录该内容

835 篇文章

订阅专栏

File "E:\project\Grounded-SAM-2-main_2.0\grounding_dino\groundingdino\models\GroundingDINO\ms_deform_attn.py", line 53, in forward
output = _C.ms_deform_attn_forward(
NameError: name '_C' is not defined

临时解决方法：

grounding_dino/groundingdino/models/GroundingDINO/ms_deform_attn.py

       if torch.cuda.is_available() and value.is_cuda and 0:
            halffloat = False
            if value.dtype == torch.float16:
                halffloat = True
                value = value.float()
                sampling_locations = sampling_locations.float()
                attention_weights = attention_weights.float()

            output = MultiScaleDeformableAttnFunction.apply(
                value,
                spatial_shapes,
                level_start_index,
                sampling_locations,
                attention_weights,
                self.im2col_step,
            )

            if halffloat:
                output = output.half()
        else:
            output = multi_scale_deformable_attn_pytorch(
                value, spatial_shapes, sampling_locations, attention_weights
            )