Не знаю, как решить эту проблему с помощью кода Python. ⇐ Python

[Anonymous]

Я использую шепот для расшифровки звука в текст. Я решил использовать distil -whisper, чтобы ускорить работу. Я пытаюсь следовать инструкциям Hugging Face, но постоянно получаю сообщение об ошибке. Я запускаю этот код последовательно, и проблем не возникает.
**

Код: Выделить всё

!pip install virtualenv
!virtualenv myenv
!myenv/bing/pip install datasets
!source myenv/bin/activate
!myenv/bin/pip install --upgrade pip
!myenv/bin/pip install --upgrade transformers accelerate datasets[audio]

**
Но потом я добавляю этот код и получаю ошибку.

Код: Выделить всё

import torch
from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
from datasets import load_dataset

device = "cuda:0" if torch.cuda.is_available() else "cpu"
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32

model_id = "distil-whisper/distil-large-v3"

model = AutoModelForSpeechSeq2Seq.from_pretrained(
model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
)
model.to(device)

processor = AutoProcessor.from_pretrained(model_id)

pipe = pipeline(
"automatic-speech-recognition",
model=model,
tokenizer=processor.tokenizer,
feature_extractor=processor.feature_extractor,
max_new_tokens=128,
torch_dtype=torch_dtype,
device=device,
)

dataset = load_dataset("distil-whisper/librispeech_long", "clean", split="validation")
sample = dataset[0]["audio"]

result = pipe(sample)
print(result["text"])

Traceback ModuleNotFoundError (последний последний вызов)
в ()
1 импорт факела
2 из трансформаторов импорт AutoModelForSpeechSeq2Seq, AutoProcessor, конвейер
----> 3 из наборов данных импорт load_dataset
4
5
ModuleNotFoundError: нет модуля с именем «наборы данных»
При установке наборов данных я не получил сообщение об ошибке, поэтому не понимаю, почему я получаю сообщение об ошибке сейчас .

Подробнее здесь: https://stackoverflow.com/questions/785 ... ython-code