0x2620/pandora_transcribe
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

expose model to manage command

Browse source
This commit is contained in:
j 2025-01-24 12:10:35 +05:30
parent 1bfa67ad5d
commit 3a680f3a1d
2 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
management/commands/transcribe.py
View file

@ -21,6 +21,8 @@ class Command(BaseCommand):
default=None, help='import into layer (default subtitle layer)') default=None, help='import into layer (default subtitle layer)')
parser.add_argument('--translate', type=str, dest='translate', parser.add_argument('--translate', type=str, dest='translate',
default='', help='list of languages to translate: (i.e. hi:en,de:en') default='', help='list of languages to translate: (i.e. hi:en,de:en')
parser.add_argument('--model', type=str, dest='model',
default='small', help='whisper model (small, large-v3-turbo, default: small)')
parser.add_argument('--gpu', action='store_true', dest='gpu', parser.add_argument('--gpu', action='store_true', dest='gpu',
default=False, help='user GPU (default: disabled)') default=False, help='user GPU (default: disabled)')
parser.add_argument('--join-sentences', action='store_true', dest='join_sentences', parser.add_argument('--join-sentences', action='store_true', dest='join_sentences',

1
transcribe.py
View file

@ -241,6 +241,7 @@ def main(**kwargs):
translate, translate,
kwargs.get("gpu"), kwargs.get("gpu"),
join_sentences=kwargs.get("join_sentences"), join_sentences=kwargs.get("join_sentences"),
model=kwargs.get("model", "small")
): ):
done.items.add(item) done.items.add(item)
queue.items.remove(item) queue.items.remove(item)