Spaces:
Build error
Build error
VanguardAI
commited on
Commit
•
61562a2
1
Parent(s):
0c6ed12
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ torch_dtype = torch.float16
|
|
14 |
# Whisper setup
|
15 |
whisper_model_id = "openai/whisper-large-v3"
|
16 |
whisper_model = AutoModelForSpeechSeq2Seq.from_pretrained(
|
17 |
-
whisper_model_id, torch_dtype=torch_dtype,
|
18 |
)
|
19 |
whisper_processor = AutoProcessor.from_pretrained(whisper_model_id)
|
20 |
whisper_pipe = pipeline(
|
@@ -27,7 +27,6 @@ whisper_pipe = pipeline(
|
|
27 |
batch_size=16,
|
28 |
return_timestamps=True,
|
29 |
torch_dtype=torch_dtype,
|
30 |
-
device=device,
|
31 |
)
|
32 |
|
33 |
# LLaMa3-8B setup
|
|
|
14 |
# Whisper setup
|
15 |
whisper_model_id = "openai/whisper-large-v3"
|
16 |
whisper_model = AutoModelForSpeechSeq2Seq.from_pretrained(
|
17 |
+
whisper_model_id, torch_dtype=torch_dtype, use_safetensors=True
|
18 |
)
|
19 |
whisper_processor = AutoProcessor.from_pretrained(whisper_model_id)
|
20 |
whisper_pipe = pipeline(
|
|
|
27 |
batch_size=16,
|
28 |
return_timestamps=True,
|
29 |
torch_dtype=torch_dtype,
|
|
|
30 |
)
|
31 |
|
32 |
# LLaMa3-8B setup
|