Update export_models.sh
Browse files- 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("./" 
     | 
| 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/ 
     | 
| 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 ./ ./ ./
         
     |