CineAI's picture
Update audio_processing/config.py
814951a verified
raw
history blame
1.11 kB
# ArcticMonkey:19.03.24:1700 example of version name in plaintext will be converted into hex using this site ->
# https://magictool.ai/tool/text-to-hex-converter/ Here ArcticMonkey is name of version and rest of all is data and time
import torch
from transformers import pipeline
device = 0 if torch.cuda.is_available() else "cpu"
checkpoint_whisper = "openai/whisper-medium"
pipe = pipeline(
"automatic-speech-recognition",
model=checkpoint_whisper,
device=device,
chunk_length_s=30,
)
# from parler_tts import ParlerTTSForConditionalGeneration
# from transformers import AutoTokenizer, AutoFeatureExtractor
# checkpoint_parler = "parler-tts/parler_tts_mini_v0.1"
# model_parler = ParlerTTSForConditionalGeneration.from_pretrained(checkpoint_parler).to(device)
# tokenizer = AutoTokenizer.from_pretrained(checkpoint_parler)
# feature_extractor = AutoFeatureExtractor.from_pretrained(checkpoint_parler)
# SAMPLE_RATE = feature_extractor.sampling_rate
# SEED = 42
# checkpoint_mms_tts_eng = "facebook/mms-tts-eng"
# pipe_tts = pipeline("text-to-speech", model=checkpoint_mms_tts_eng)