Browse Source

Remove local only that is failing dora-distil-whishper

tags/v0.3.7rc0
haixuanTao 1 year ago
parent
commit
07a991bfa0
1 changed files with 1 additions and 5 deletions
  1. +1
    -5
      node-hub/dora-distil-whisper/dora_distil_whisper/main.py

+ 1
- 5
node-hub/dora-distil-whisper/dora_distil_whisper/main.py View File

@@ -10,11 +10,7 @@ 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,
local_files_only=True,
model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
)
model.to(device)



Loading…
Cancel
Save