dac202 commited on
Commit
ab759b9
·
1 Parent(s): ad16150

fixed autoexec sequence

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def load_whisper_model(model_path, lora_config, base_model_name="openai/whisper-
43
  model = PeftModel.from_pretrained(model, lora_config)
44
 
45
  mdoel = model.merge_and_unload()
46
- model.save_pretrained(model_path)
47
 
48
  print(f'Whisper model from {lora_config} saved at {model_path}')
49
  return
 
43
  model = PeftModel.from_pretrained(model, lora_config)
44
 
45
  mdoel = model.merge_and_unload()
46
+ model.save_pretrained(model_path, save_serialization=False)
47
 
48
  print(f'Whisper model from {lora_config} saved at {model_path}')
49
  return