@@ -3065,7 +3065,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
30653065 )
30663066 quantizer = PostTrainingQuantizer (model , quantization_config )
30673067 model = cast ("nn.Module" , quantizer .compress ())
3068- transform_with_custom_compression_ops (model )
3068+ transform_with_custom_compression_ops (model ) # noqa: F821
30693069
30703070 sample = (torch .randn (2 , 24 ),)
30713071
@@ -3153,7 +3153,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
31533153 )
31543154 quantizer = PostTrainingQuantizer (model , quantization_config )
31553155 model = cast ("nn.Module" , quantizer .compress ())
3156- transform_with_custom_compression_ops (model )
3156+ transform_with_custom_compression_ops (model ) # noqa: F821
31573157
31583158 sample = (torch .randn (2 , 24 ),)
31593159
@@ -3217,7 +3217,7 @@ def forward(self, x: torch.Tensor, indices: torch.Tensor) -> torch.Tensor:
32173217 )
32183218 quantizer = PostTrainingQuantizer (model , quantization_config )
32193219 model = cast ("nn.Module" , quantizer .compress ())
3220- transform_with_custom_compression_ops (model )
3220+ transform_with_custom_compression_ops (model ) # noqa: F821
32213221
32223222 x = torch .randn (2 , 1 , 1 , in_dim )
32233223 indices = torch .tensor ([[0 , 2 ], [1 , 3 ]], dtype = torch .int16 )
@@ -3318,7 +3318,7 @@ def forward(self, x: torch.Tensor, indices: torch.Tensor) -> torch.Tensor:
33183318 )
33193319 quantizer = PostTrainingQuantizer (model , quantization_config )
33203320 model = cast ("nn.Module" , quantizer .compress ())
3321- transform_with_custom_compression_ops (model )
3321+ transform_with_custom_compression_ops (model ) # noqa: F821
33223322
33233323 x = torch .randn (2 , 1 , 1 , in_dim )
33243324 indices = torch .tensor ([[0 , 2 ], [1 , 3 ]], dtype = torch .int16 )
@@ -3390,7 +3390,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
33903390 )
33913391 quantizer = PostTrainingQuantizer (model , quantization_config )
33923392 model = cast ("nn.Module" , quantizer .compress ())
3393- transform_with_custom_compression_ops (model )
3393+ transform_with_custom_compression_ops (model ) # noqa: F821
33943394
33953395 sample = (torch .randn (1 , 4 , embed_dim ),)
33963396
@@ -3484,7 +3484,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
34843484 )
34853485 quantizer = PostTrainingQuantizer (model , quantization_config )
34863486 model = cast ("nn.Module" , quantizer .compress ())
3487- transform_with_custom_compression_ops (model )
3487+ transform_with_custom_compression_ops (model ) # noqa: F821
34883488
34893489 sample = (torch .randn (1 , 4 , embed_dim ),)
34903490
@@ -3553,7 +3553,7 @@ def forward(self, x: torch.Tensor, indices: torch.Tensor) -> torch.Tensor:
35533553 )
35543554 quantizer = PostTrainingQuantizer (model , quantization_config )
35553555 model = cast ("nn.Module" , quantizer .compress ())
3556- transform_with_custom_compression_ops (model )
3556+ transform_with_custom_compression_ops (model ) # noqa: F821
35573557
35583558 x = torch .randn (2 , in_dim )
35593559 indices = torch .tensor ([[0 , 2 ], [1 , 3 ]], dtype = torch .int16 )
@@ -3637,7 +3637,7 @@ def forward(self, x: torch.Tensor, indices: torch.Tensor) -> torch.Tensor:
36373637 )
36383638 quantizer = PostTrainingQuantizer (model , quantization_config )
36393639 model = cast ("nn.Module" , quantizer .compress ())
3640- transform_with_custom_compression_ops (model )
3640+ transform_with_custom_compression_ops (model ) # noqa: F821
36413641
36423642 x = torch .randn (2 , in_dim )
36433643 indices = torch .tensor ([[0 , 2 ], [1 , 3 ]], dtype = torch .int16 )
0 commit comments