Skip to content

Commit a427dee

Browse files
Update packages/tasks/src/tasks/audio-text-to-text/about.md
Co-authored-by: Pedro Cuenca <[email protected]>
1 parent cdeaba0 commit a427dee

File tree

1 file changed

+1
-1
lines changed
  • packages/tasks/src/tasks/audio-text-to-text

1 file changed

+1
-1
lines changed

packages/tasks/src/tasks/audio-text-to-text/about.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ device = "cuda"
3434
repo_id = "mistralai/Voxtral-Mini-3B-2507"
3535

3636
processor = AutoProcessor.from_pretrained(repo_id)
37-
model = VoxtralForConditionalGeneration.from_pretrained(repo_id, torch_dtype=torch.bfloat16, device_map=device)
37+
model = VoxtralForConditionalGeneration.from_pretrained(repo_id, dtype=torch.bfloat16, device_map=device)
3838

3939
inputs = processor.apply_transcription_request(language="en", audio="https://huggingface.co/datasets/hf-internal-testing/dummy-audio-samples/resolve/main/obama.mp3", model_id=repo_id)
4040
inputs = inputs.to(device, dtype=torch.bfloat16)

0 commit comments

Comments
 (0)