Skip to content

Commit ec4cbc9

Browse files
committed
fixed tests
1 parent 85cea14 commit ec4cbc9

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/torch/quantization/test_strip.py

+2
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
from nncf.config import NNCFConfig
2626
from nncf.parameters import CompressWeightsMode
2727
from nncf.parameters import StripFormat
28+
from nncf.quantization.advanced_parameters import AdvancedCompressionParameters
2829
from nncf.torch.graph.transformations.commands import ExtraCompressionModuleType
2930
from nncf.torch.quantization.layers import AsymmetricQuantizer
3031
from nncf.torch.quantization.layers import BaseQuantizer
@@ -380,6 +381,7 @@ def test_nncf_strip_lora_model(mode, decompressor_class, torch_dtype, atol, mock
380381
mode=mode,
381382
dataset=nncf.Dataset(dataset),
382383
compression_format=nncf.CompressionFormat.FQ_LORA,
384+
advanced_parameters=AdvancedCompressionParameters(lora_adapter_rank=1),
383385
)
384386
if mode in [CompressWeightsMode.INT4_SYM, CompressWeightsMode.INT4_ASYM]:
385387
compression_kwargs.update(dict(ratio=1, group_size=4, all_layers=True))

0 commit comments

Comments
 (0)