Spaces:
Sleeping
Sleeping
fixed autoexec sequence
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ def load_whisper_model(model_path, lora_config, base_model_name="openai/whisper-
|
|
42 |
|
43 |
model = PeftModel.from_pretrained(model, lora_config)
|
44 |
|
45 |
-
|
46 |
model.save_pretrained(model_path, save_serialization=False)
|
47 |
|
48 |
print(f'Whisper model from {lora_config} saved at {model_path}')
|
|
|
42 |
|
43 |
model = PeftModel.from_pretrained(model, lora_config)
|
44 |
|
45 |
+
model = 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}')
|