pere commited on
Commit
37ab78c
1 Parent(s): bf871bf

Update export_models.sh

Browse files
Files changed (1) hide show
  1. export_models.sh +2 -2
export_models.sh CHANGED
@@ -16,7 +16,7 @@ model.save_pretrained("./")
16
  print("Done.")
17
 
18
  print("Saving model to TensorFlow...", end=" ")
19
- tf_model = TFWhisperForConditionalGeneration.from_pretrained("./", from_pt=True)
20
  tf_model.save_pretrained("./")
21
  print("Done.")
22
 
@@ -41,7 +41,7 @@ print("Done")
41
  END
42
 
43
  echo "Saving model to GGML (whisper.cpp)..."
44
- wget -O convert-h5-to-ggml.py "https://raw.githubusercontent.com/ggerganov/whisper.cpp/94aa56f19eed8b2419bc5ede6b7fda85d5ca59be/models/convert-h5-to-ggml.py"
45
  mkdir -p whisper/assets
46
  wget -O whisper/assets/mel_filters.npz "https://github.com/openai/whisper/raw/c5d42560760a05584c1c79546a098287e5a771eb/whisper/assets/mel_filters.npz"
47
  python ./convert-h5-to-ggml.py ./ ./ ./
 
16
  print("Done.")
17
 
18
  print("Saving model to TensorFlow...", end=" ")
19
+ tf_model = TFWhisperForConditionalGeneration.from_pretrained("./")
20
  tf_model.save_pretrained("./")
21
  print("Done.")
22
 
 
41
  END
42
 
43
  echo "Saving model to GGML (whisper.cpp)..."
44
+ wget -O convert-h5-to-ggml.py "https://raw.githubusercontent.com/NbAiLab/nb-whisper/main/convert-h5-to-ggml.py"
45
  mkdir -p whisper/assets
46
  wget -O whisper/assets/mel_filters.npz "https://github.com/openai/whisper/raw/c5d42560760a05584c1c79546a098287e5a771eb/whisper/assets/mel_filters.npz"
47
  python ./convert-h5-to-ggml.py ./ ./ ./