update 1.0.0.post2

2024-01-19 22:18:59 +08:00
parent 7d59305b5f
commit 6b3e1d6fe7
9 changed files with 21 additions and 15 deletions
--- a/DCNv4_op/DCNv4/functions/flash_deform_attn_func.py
+++ b/DCNv4_op/DCNv4/functions/flash_deform_attn_func.py
@@ -57,7 +57,7 @@ def findspec_bwd(B, Q, G, C):
        d_stride = 2
    else:
        d_stride = 1
-    
+
    ms = factors(B*Q)
    multiplier = 1
    for m in ms:
@@ -70,7 +70,7 @@ class FlashDeformAttnFunction(Function):
    @staticmethod
    @torch.autocast("cuda", enabled=True, dtype=torch.float16)
    def forward(
-        ctx, value, value_spatial_shapes, value_level_start_index, 
+        ctx, value, value_spatial_shapes, value_level_start_index,
        sampling_loc_attn, im2col_step, K=8
    ):