Skip to content

Commit 6729c6e

Browse files
committed
πŸ†™ Update version
1 parent e27f4dd commit 6729c6e

File tree

2 files changed

+10
-6
lines changed

2 files changed

+10
-6
lines changed

β€ŽREADME.mdβ€Ž

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,15 @@ from transformers import BitsAndBytesConfig, HqqConfig
3939
url = "https://www.youtube.com/watch?v=di3rHkEZuUw"
4040
audio_path = download_and_convert_to_mp3(url)
4141

42-
quant_config = HqqConfig(
43-
nbits=1, group_size=64, quant_zero=False, quant_scale=False, axis=0
42+
hqq_config = HqqConfig(
43+
nbits=1,
44+
group_size=64,
45+
quant_zero=False,
46+
quant_scale=False,
47+
axis=0,
48+
offload_meta=False,
4449
) # axis=0 is used by default
4550

46-
4751
bnb_config = BitsAndBytesConfig(
4852
load_in_4bit=True,
4953
bnb_4bit_quant_type="nf4",
@@ -52,11 +56,11 @@ bnb_config = BitsAndBytesConfig(
5256
)
5357

5458
pipeline = SpeechToTextPipeline(
55-
model_id="distil-whisper/distil-large-v3", quant_config=quant_config
59+
model_id="distil-whisper/distil-large-v3", quant_config=hqq_config
5660
) # or bnb_config
5761

5862
transcript = pipeline(
59-
audio_path="test.mp3",
63+
audio_path=audio_path,
6064
chunk_length_s=30,
6165
stride_length_s=5,
6266
max_new_tokens=128,

β€Žwhisperplus/__init__.pyβ€Ž

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,6 @@
77
from whisperplus.utils.download_utils import download_and_convert_to_mp3
88
from whisperplus.utils.text_utils import format_speech_to_dialogue
99

10-
__version__ = '0.2.7.2.dev1'
10+
__version__ = '0.3.0'
1111
__author__ = 'kadirnar'
1212
__license__ = 'Apache License 2.0'

0 commit comments

Comments
Β (0)