Skip to content

Commit 9542bc9

Browse files
fix: warnings related to chunk_length_s in ifw (#258)
1 parent 5e7e561 commit 9542bc9

2 files changed

Lines changed: 1 addition & 1 deletion

File tree

src/pruna/algorithms/batching/ifw.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,7 @@ def _apply(self, model: Any, smash_config: SmashConfigPrefixWrapper) -> Any:
128128
else {"attn_implementation": "sdpa"}
129129
),
130130
device=smash_config["device"],
131+
ignore_warning=True,
131132
)
132133
return pipe
133134

tests/fixtures.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,6 @@ def whisper_tiny_random_model() -> tuple[Any, SmashConfig]:
6262
model = pipeline(
6363
"automatic-speech-recognition",
6464
model=model_id,
65-
chunk_length_s=30,
6665
torch_dtype=torch.float16,
6766
device_map="cpu",
6867
)

0 commit comments

Comments
 (0)